↓ 1 callersMethod__first_step_init__(self,
lr=1e-3, bias_correction = True, grad_averaging=True,
betas=(0.9, 0.9
segmentation/apex/apex/contrib/optimizers/distributed_fused_lamb.py:106
↓ 1 callersMethod__init__(self, block, layers, num_classes=1000,
groups=1, width_per_group=64, norm_layer=None)
segmentation/apex/apex/pyprof/examples/user_annotation/resnet.py:104
↓ 1 callersMethod__launch_step_kernel(self, p, p_copy, m, v, g)
segmentation/apex/apex/contrib/optimizers/distributed_fused_adam_v3.py:218
↓ 1 callersMethod_check_branches(self, num_branches, blocks, num_blocks,
num_inchannels, num_channels)
segmentation/mseg-semantic/mseg_semantic/model/seg_hrnet.py:139
↓ 1 callersMethod_make_one_branch(self, branch_index, block, num_blocks, num_channels,
stride=1)
segmentation/mseg-semantic/mseg_semantic/model/seg_hrnet.py:159
↓ 1 callersFunctionaxpby_check_overflow_python(model_grad, stashed_grad, master_grad, a, b, check_overflow=False)
segmentation/apex/apex/amp/scaler.py:19