it hangs at loading the pretrained model and here is the call stack.
torch==1.7.1
Any idea on how to fix it? Thanks
Traceback (most recent call first):
  File "/opt/conda/lib/python3.7/site-packages/apex/amp/_initialize.py", line 141, in __call__
    param = param.to(torch.float32)
  File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 909, in state_dict
    hook_result = hook(self, destination, prefix, local_metadata)
  File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 907, in state_dict
    module.state_dict(destination, prefix + name + '.', keep_vars=keep_vars)
  File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 907, in state_dict
    module.state_dict(destination, prefix + name + '.', keep_vars=keep_vars)
  File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 907, in state_dict
    module.state_dict(destination, prefix + name + '.', keep_vars=keep_vars)
  File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 907, in state_dict
    module.state_dict(destination, prefix + name + '.', keep_vars=keep_vars)
  File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 907, in state_dict
    module.state_dict(destination, prefix + name + '.', keep_vars=keep_vars)
  File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 907, in state_dict
    module.state_dict(destination, prefix + name + '.', keep_vars=keep_vars)
  File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 907, in state_dict
    module.state_dict(destination, prefix + name + '.', keep_vars=keep_vars)
  File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 907, in state_dict
    module.state_dict(destination, prefix + name + '.', keep_vars=keep_vars)
  (frame information optimized out)
  (frame information optimized out)
  File "/tmp/code/quickdetection/src/qd/opt/checkpoint.py", line 267, in load
    self._load_model(checkpoint)
  File "/tmp/code/quickdetection/src/qd/opt/checkpoint.py", line 249, in recover_or_load
    return self.load(f, model_only, load_if_has)
  (frame information optimized out)
  (frame information optimized out)
  File "/tmp/code/quickdetection/src/qd/pipeline.py", line 674, in pipeline_train_eval_multi
    pip.ensure_train()
  File "src/qd/qd_common.py", line 3351, in execute_func
  File "src/qd/qd_common.py", line 3827, in <module>