From 9ae5fa1a88899180f88ddb14d4cef457ceb847e5 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 5 Apr 2021 20:47:55 +0200 Subject: Add new training loop with PyTorch Lightning, remove stale files --- training/sweep_emnist.yml | 26 -------------------------- 1 file changed, 26 deletions(-) delete mode 100644 training/sweep_emnist.yml (limited to 'training/sweep_emnist.yml') 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 -- cgit v1.2.3-70-g09d2