Update train.py - Githubissues

OpenGVLab / InternImage

[CVPR 2023 Highlight] InternImage: Exploring Large-Scale Vision Foundation Models with Deformable Convolutions

https://arxiv.org/abs/2211.05778

MIT License

2.48k stars 231 forks source link

Update train.py #174

Closed SmartOinker closed 1 year ago

SmartOinker commented 1 year ago

Running the orignial train.py with dist_train.sh will lead to the error: unrecognized arguments: --local_rank=0 on pytorch 2. This edit will fix the error.