chenxin061 / pdarts

Codes for our paper "Progressive Differentiable Architecture Search:Bridging the Depth Gap between Search and Evaluation"
Other
360 stars 83 forks source link

如果只有GTX1080ti(11G内存),如何调整参数,让pdarts也可以跑起来训练? #2

Closed simonggx closed 5 years ago

198808xc commented 5 years ago

Hello,

First of all, I suggest you to modify the title into English, so that more people can benefit from this post.

If you do have a GPU with smaller memory, you can modify the batch size. In your case, from 96 to 64 in CIFAR10/100. The speed will be a bit slower, but the accuracy would not be largely impacted.

We would post more results with this setting in README.

Best, Authors