yuhuixu1993 / PC-DARTS

PC-DARTS:Partial Channel Connections for Memory-Efficient Differentiable Architecture Search
434 stars 108 forks source link

About "replace input_search, target_search = next(iter(valid_queue))" #34

Open swlzq opened 4 years ago

swlzq commented 4 years ago

Thank you for your release code! I noticed that you replaced "input_search, target_search = next(iter(valid_queue))", why it is much faster? And why the code in "try" is "next(valid_queue_iter)" instead of "next(valid_queue)"? Hope for your reply! try: input_search, target_search = next(valid_queue_iter) except: valid_queue_iter = iter(valid_queue) input_search, target_search = next(valid_queue_iter)