Swin transformer TypeError: __init__() got an unexpected keyword argument ‘t_mul‘

Traceback (most recent call last):
  File "main.py", line 357, in 
    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 
    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.,’

你可能感兴趣的:(transformer,深度学习,人工智能)