BiSeNet icon indicating copy to clipboard operation
BiSeNet copied to clipboard

Training errow

Open Jasonx0123 opened this issue 3 years ago • 2 comments
trafficstars

2a7bd65854f0a80d585a76f7e410bc2 How to solve this problem?

Jasonx0123 avatar Aug 22 '22 04:08 Jasonx0123

Please follow the step of readme to launch training, and please use same version of pytorch.

CoinCheung avatar Aug 23 '22 12:08 CoinCheung

Thanks for your reply, I didn't change the version of pytorch. I add this code into train_amp.py and the problem was solved. parse.add_argument('--local_rank', defautle=-1, type=int, help='node rank for distributed training')

Jasonx0123 avatar Aug 24 '22 08:08 Jasonx0123

I am closing this, since the problem is solved.

CoinCheung avatar Nov 15 '22 03:11 CoinCheung