Add new training loop with PyTorch Lightning, remove stale files

1 files changed, 0 insertions, 26 deletions

diff --git a/training/sweep_emnist.yml b/training/sweep_emnist.yml
deleted file mode 100644
index 48d7261..0000000
--- a/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