search

microsoft / Swin-Transformer

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
https://arxiv.org/abs/2103.14030
MIT License
13.96k stars 2.06k forks source link

train #324

Open 15812853258 opened 1 year ago

15812853258 commented 1 year ago

How to modify the code to train on a single card?Instead of using distributed training?

RohollahHS commented 1 year ago

Hi, Have you found a way to train Swin Transformer on a local machine?

YoRsk commented 1 year ago

same of mine.

RohollahHS commented 1 year ago

Actually I managed to run the code on my machine. You just have to remove codes that are realted to disributed programming and it works fine.