microsoft / nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
https://nni.readthedocs.io
MIT License
13.88k stars 1.81k forks source link

How to load model in test dataset #5795

Open Marks00996 opened 1 week ago

Marks00996 commented 1 week ago

In the DARTS tutorial, I replaced Cifar10 with my own dataset and the val results were good, but the Acc performance on my custom test.py file was very poor. I think there is a problem with model loading. Could you tell me how to load the model correctly and run it on the test dataset? Retrain result: image test result image Help me!!!!