D-X-Y / AutoDL-Projects

Automated deep learning algorithms implemented in PyTorch.
MIT License
1.56k stars 281 forks source link

Question about NAS Acceleration #4

Closed yanwei-li closed 5 years ago

yanwei-li commented 5 years ago

Hi! Thanks for your good job!

I have some questions about the Fast Acceleration version. In the main paper, it is said to adopt Gumbel Softmax with temperature to relax argmax function during the searching process. But in "model_search.py", I found it using the normal version which calculates all ops' probs (same with that in DARTS).

Is it something wrong or just my fault?

D-X-Y commented 5 years ago

@yanwei-li Thanks for your interest. model_search.py is the compared algorithm (DARTS first order)of our GDAS. We did not include our searching model in this repo.