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
smeenapadnekar
commented
7 months ago
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