BasicSR icon indicating copy to clipboard operation
BasicSR copied to clipboard

train.py: error: unrecognized arguments: --local-rank=0

Open jere1882 opened this issue 2 years ago • 1 comments

unable to run train.py, I get error

train.py: error: unrecognized arguments: --local-rank=0

I'm not manually passing this parameter.

jere1882 avatar Jun 04 '23 15:06 jere1882

You need to change your PyTorch version from 2.0 to 1.x because the launch.py in 2.0 uses 'local_rank' instead of 'local-rank'.

SecureSheII avatar Jun 13 '23 23:06 SecureSheII