facebookresearch / facebookresearch/moco

Unable to load pre-trained Moco V2 model

Open
#89 1 comment 7 reactions 0 assignees View on GitHub
Dominant language
No language data
Stars
5.1k
Forks
802
PR merge metrics
No merged PRs in 30d

Description

When attempting to load the pre-trained Moco V2 model, trained on 800 epochs (https://dl.fbaipublicfiles.com/moco/moco_checkpoints/moco_v2_800ep/moco_v2_800ep_pretrain.pth.tar), I am getting the following error message. Please advise.

Use GPU: 0 for training
Traceback (most recent call last):
File "main_moco_rtog.py", line 521, in
main()
File "main_moco_rtog.py", line 222, in main
mp.spawn(main_worker, nprocs=ngpus_per_node, args=(ngpus_per_node, args, paths))
File "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/torch/multiprocessing/spawn.py", line 200, in spawn
return start_processes(fn, args, nprocs, join, daemon, start_method='spawn')
File "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/torch/multiprocessing/spawn.py", line 158, in start_processes
while not context.join():
File "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/torch/multiprocessing/spawn.py", line 119, in join
raise Exception(msg)
Exception:

-- Process 1 terminated with the following error:
Traceback (most recent call last):
File "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/torch/multiprocessing/spawn.py", line 20, in _wrap
fn(i, *args)
File "/export/home/code/metamind/precision_oncology/SSL/moco/main_moco_rtog.py", line 275, in main_worker
assert set(msg.missing_keys) == {"fc.weight", "fc.bias"}
AssertionError

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.