diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-03-20 18:09:06 +0100 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-03-20 18:09:06 +0100 |
commit | 7e8e54e84c63171e748bbf09516fd517e6821ace (patch) | |
tree | 996093f75a5d488dddf7ea1f159ed343a561ef89 /src/training/sweep_emnist.yml | |
parent | b0719d84138b6bbe5f04a4982dfca673aea1a368 (diff) |
Inital commit for refactoring to lightning
Diffstat (limited to 'src/training/sweep_emnist.yml')
-rw-r--r-- | src/training/sweep_emnist.yml | 26 |
1 files changed, 0 insertions, 26 deletions
diff --git a/src/training/sweep_emnist.yml b/src/training/sweep_emnist.yml deleted file mode 100644 index 48d7261..0000000 --- a/src/training/sweep_emnist.yml +++ /dev/null @@ -1,26 +0,0 @@ -program: training/run_sweep.py -method: bayes -metric: - name: val_loss - goal: minimize -parameters: - dataset: - value: EmnistDataset - model: - value: CharacterModel - network: - value: MLP - network_args.hidden_size: - values: [128, 256] - network_args.dropout_rate: - values: [0.2, 0.4] - network_args.num_layers: - values: [3, 6] - optimizer_args.lr: - values: [1.e-1, 1.e-5] - lr_scheduler_args.max_lr: - values: [1.0e-1, 1.0e-5] - train_args.batch_size: - values: [64, 128] - train_args.epochs: - value: 5 |