summaryrefslogtreecommitdiff
path: root/src/training/sweep_emnist.yml
diff options
context:
space:
mode:
Diffstat (limited to 'src/training/sweep_emnist.yml')
-rw-r--r--src/training/sweep_emnist.yml26
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