def __init__()

in benchmarks/horovod-resnet/train_imagenet_resnet_hvd.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, optimizer, scale=None, name="MixedPrecisionOptimizer", use_locking=False):
        super(MixedPrecisionOptimizer, self).__init__(name=name, use_locking=use_locking)
        self._optimizer = optimizer
        self._scale = float(scale) if scale is not None else 1.0