how to train by using multi GPU?

Question

Opened this issue 5 years ago · 3 comments

Hi:
I have 8 GPUs in one computer. if I use model = nn.DataParallel(model) in train_serach.py，it can't works.How to train pc-dart by using 8 gpus?

Answer 1 · 2020-02-25T16:02:38.000Z

Hi, @sybilWsybil ,I think there is no need to use 8 gpus on CIFAR. You can run train_search_imagenet.py to search on Imagenet.

Answer 2 · 2020-02-27T08:23:22.000Z

Thank you!

Answer 3 · 2020-09-22T08:55:41.000Z

你好，请问运行test.py文件时，里边的model.pt是什么时候生成的，为什么我运行了train_search.py和train.py都没有生成这个文件？