From 3061eebd2dba05f73cbd2f1749885da933896e07 Mon Sep 17 00:00:00 2001
From: wondergo2017 <wondergo2017@gmail.com>
Date: Wed, 8 Jun 2022 20:50:13 +0800
Subject: [PATCH] fix train scratch estimator

---
 autogl/module/nas/estimator/train_scratch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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,
         )