Transformer-SSL icon indicating copy to clipboard operation
Transformer-SSL copied to clipboard

Published 20 hours ago • verified publisher icon SwinTransformer

Train MoBy-SwinT on local machine with one GPU

Open TNA8 opened this issue 4 years ago • 2 comments
trafficstars

I am gonna train MoBY-SwinT on my custom dataset. My machine has one GPU. I tried some but failed and faced following errors. All packages are installed.

  • First try screenshot_33

  • Second try screenshot_34

What is the correct command to run the training script on local machine with one GPU?

Thanks in advance.

TNA8 avatar Nov 12 '21 11:11 TNA8

Try "torch.distributed", not "torch.distrubuted"

notabigfish avatar Feb 21 '22 10:02 notabigfish

I have a similar problem, have you solved it?

L-xiaolu avatar Jun 23 '22 09:06 L-xiaolu