capstone2019-neuralsearch / AC297r_2019_NAS

Harvard IACS Data Science Capstone: Neural Architecture Search (NAS) with Google
5 stars 1 forks source link

DARTS on Graphene #6

Closed dylanrandle closed 4 years ago

dylanrandle commented 4 years ago

Integrate graphene dataset to use with DARTS

JiaweiZhuang commented 4 years ago

Based on #3, I think the only change for DARTS is to use nn.MSELoss() instead of nn.CrossEntropyLoss()