Problem reloading weights
Closed this issue · 0 comments
TimZaman commented
2019-01-06 18:42:18,355 INFO Downloading: runs/Jan06_04-11-46_optimizer-master-0/model_000000726.pt
Traceback (most recent call last):
File "optimizer.py", line 362, in <module>
pretrained_model=args.pretrained_model,
File "optimizer.py", line 338, in main
pretrained_model=pretrained_model,
File "optimizer.py", line 80, in __init__
self.policy_base.load_state_dict(torch.load(pretrained_model), strict=False)
File "/root/.local/lib/python3.7/site-packages/torch/serialization.py", line 367, in load
return _load(f, map_location, pickle_module)
File "/root/.local/lib/python3.7/site-packages/torch/serialization.py", line 528, in _load
magic_number = pickle_module.load(f)
EOFError: Ran out of input