diff --git a/nn/conf/config.yaml b/nn/conf/config.yaml index 0db3fe6..31bce6b 100644 --- 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 index 8029b84..9edecc4 100644 --- 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")