↓ 6 callersMethodunscale(self, model_grads, master_grads, unused_scale, models_are_masters=False, scale_override=None)
segmentation/apex/apex/amp/scaler.py:94
↓ 4 callersMethod_make_layer(self, block, planes, blocks, stride=1, dilate=False)
segmentation/apex/apex/pyprof/examples/user_annotation/resnet.py:134
↓ 4 callersMethodconv_bytes_flops(self, N, C, H, W, K, P, Q, R, S, g, t)
segmentation/apex/apex/pyprof/prof/conv.py:184