using kaiming init in resnet

2021-02-08 16:47:38 +08:00 · 2021-02-08 16:47:38 +08:00 · b95f9fdb8e
parent 506ab6f477
commit b95f9fdb8e
1 changed files with 0 additions and 2 deletions
--- a/model_zoo/official/cv/resnet/src/resnet.py
+++ b/model_zoo/official/cv/resnet/src/resnet.py
@ -74,8 +74,6 @@ def _calculate_fan_in_and_fan_out(tensor):
        num_output_fmaps = tensor[0]
        receptive_field_size = 1
        if dimensions > 2:
-            import time
-            time.sleep(10)
            receptive_field_size = tensor[2] * tensor[3]
        fan_in = num_input_fmaps * receptive_field_size
        fan_out = num_output_fmaps * receptive_field_size