What's the required hardware to reproduce the result?

YyzHarry / imbalanced-semi-self

[NeurIPS 2020] Semi-Supervision (Unlabeled Data) & Self-Supervision Improve Class-Imbalanced / Long-Tailed Learning

https://arxiv.org/abs/2006.07529

MIT License

736 stars 116 forks source link

What's the required hardware to reproduce the result? #8

Closed zjamy closed 4 years ago

zjamy commented 4 years ago

Thanks for sharing this code. It's interesting. May I know the required hardware to reproduce the result?

The reason I'm asking because I tried to run "pretrain_rot.py --dataset 'cifar10' --imb_factor 0.01 ", but the system doesn't response for a long time when running at "output = model(inputs)".

zjamy commented 4 years ago

I fix it when I set "--gpu 0".