Train MoBy-SwinT on local machine with one GPU

SwinTransformer / Transformer-SSL

This is an official implementation for "Self-Supervised Learning with Swin Transformers".

https://arxiv.org/abs/2105.04553

MIT License

624 stars 67 forks source link

Train MoBy-SwinT on local machine with one GPU #11

Open TNA8 opened 2 years ago

TNA8 commented 2 years ago

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
Second try

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

Thanks in advance.

notabigfish commented 2 years ago

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

L-xiaolu commented 2 years ago

I have a similar problem, have you solved it?