Traceback (most recent call last):
File "main.py", line 357, in <module>
main(config)
File "main.py", line 99, in main
lr_scheduler = build_scheduler(config, optimizer, len(data_loader_train))
File "/home/ubuntu/wgp/Swin-Transformer/lr_scheduler.py", line 21, in build_scheduler
lr_scheduler = CosineLRScheduler(
TypeError: __init__() got an unexpected keyword argument 't_mul'
Killing subprocess 19306
Traceback (most recent call last):
File "/opt/anaconda3/envs/swin/lib/python3.8/runpy.py", line 194, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/opt/anaconda3/envs/swin/lib/python3.8/runpy.py", line 87, in _run_code
exec(code, run_globals)
File "/opt/anaconda3/envs/swin/lib/python3.8/site-packages/torch/distributed/launch.py", line 340, in <module>
main()
File "/opt/anaconda3/envs/swin/lib/python3.8/site-packages/torch/distributed/launch.py", line 326, in main
sigkill_handler(signal.SIGTERM, None) # not coming back
File "/opt/anaconda3/envs/swin/lib/python3.8/site-packages/torch/distributed/launch.py", line 301, in sigkill_handler
raise subprocess.CalledProcessError(returncode=last_return_code, cmd=cmd)
subprocess.CalledProcessError: Command '['/opt/anaconda3/envs/swin/bin/python3.8', '-u', 'main.py', '--local_rank=0', '--eval', '--cfg', 'configs/swin_tiny_patch4_window7_224.yaml', '--resume', '/pth/swin_tiny_patch4_window7_224.pth', '--data-path', 'imagenet']' returned non-zero exit status 1.
解决办法，删除Swin-Transformer/lr_scheduler.py的第24行‘t_mul=1.,’
|