diff --git a/autogl/module/nas/estimator/train_scratch.py b/autogl/module/nas/estimator/train_scratch.py index 84111c1..9c448d4 100644 --- a/autogl/module/nas/estimator/train_scratch.py +++ b/autogl/module/nas/estimator/train_scratch.py @@ -36,7 +36,7 @@ class TrainEstimator(BaseEstimator): weight_decay=5e-4, device="auto", init=False, - feval=self.evaluation, + feval='acc', loss=self.loss_f, lr_scheduler_type=None, )