search

khanrc / pt.darts

PyTorch Implementation of DARTS: Differentiable Architecture Search
MIT License
439 stars 108 forks source link

Search Not Working? #33

Open rpand002 opened 4 years ago

rpand002 commented 4 years ago

Hi @khanrc, Thanks for releasing the code. I am trying to search a cell using DARTS as mentioned in the README. However, the search is not working. It becomes stuck at the very beginning (training seems to be not working). There is no GPU utilization either. Am I missing something here? Can you please comment on this? Thanks.

Screen Shot 2020-02-11 at 2 09 29 AM
Frizy-up commented 4 years ago

Set DataLoader pin_memory=False, it working. It seems to be stuck due to memory error.

rpand002 commented 4 years ago

Thanks @Frizy-up. It works now. However the search seems to be very slow on a single TITAN GPU. It takes about 4 minutes for 50 iterations/steps (both for pin_memory=True and pin_memory=False). Any comment on this.

Frizy-up commented 4 years ago

Thanks @Frizy-up. It works now. However the search seems to be very slow on a single TITAN GPU. It takes about 4 minutes for 50 iterations/steps (both for pin_memory=True and pin_memory=False). Any comment on this.

That's well, The time cost is same as the paper reported.

weihui98 commented 4 years ago

Hello, Have you solved this problem? I change the "pin_memory" to "False", while it still does not work......

Thx