!12284 bug fix: the resnet initial time increase after use kaiming initial

From: @sl_wang Reviewed-by: @wang_zi_dong,@kisnwang Signed-off-by: @kisnwang
2021-02-08 19:27:38 +08:00 · 2021-02-08 19:27:38 +08:00 · e8060bd9e8
parent 0ce5faf3f6 b95f9fdb8e
commit e8060bd9e8
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