quark0 / darts

Differentiable architecture search for convolutional and recurrent networks
https://arxiv.org/abs/1806.09055
Apache License 2.0
3.91k stars 845 forks source link

Loading the model after training #173

Open Tahmid1406 opened 3 weeks ago

Tahmid1406 commented 3 weeks ago

After the search process, ("using the train_search") I end up with a weights.pt file, a log file, reduction and normal cells.

My question is how do I load the best model and do inference on it for the cifar10 model?