Browse Source

fix train scratch estimator

develop/0.4/predevelop
wondergo2017 4 years ago
parent
commit
3061eebd2d
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      autogl/module/nas/estimator/train_scratch.py

+ 1
- 1
autogl/module/nas/estimator/train_scratch.py View File

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


Loading…
Cancel
Save