Bug fix

commanderxa · Oct 8, 2023 · 02268ae · 02268ae
1 parent 9a1f90b
commit 02268ae
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nn/conf/config.yaml b/nn/conf/config.yaml
@@ -20,7 +20,7 @@ hyper:
   n_hidden: 256
   batch_size: 128
   dropout: 0.2
-  lr: 3e-4
+  lr: 3e-3
   epochs: 10
   eval_iters: 1
   use_amp: False

diff --git a/nn/train.py b/nn/train.py
@@ -16,7 +16,7 @@
 from dscovry.dataset import DSCOVRDataset
 from dscovry.model import DSCOVRYModel
 from dscovry.config import Config
-from .utils import evaluate_accuracy
+from utils import evaluate_accuracy
 
 
 @hydra.main(version_base=None, config_path="conf", config_name="config")