chenxin061 / pdarts

Codes for our paper "Progressive Differentiable Architecture Search:Bridging the Depth Gap between Search and Evaluation"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

simonggx opened this issue · comments

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