From 7268035fb9e57342612a8cc50a1fe04e8841ca2f Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Fri, 30 Jul 2021 23:15:03 +0200 Subject: attr bug fix, properly loading network --- training/conf/experiment/vqvae_experiment.yaml | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 training/conf/experiment/vqvae_experiment.yaml (limited to 'training/conf/experiment/vqvae_experiment.yaml') diff --git a/training/conf/experiment/vqvae_experiment.yaml b/training/conf/experiment/vqvae_experiment.yaml new file mode 100644 index 0000000..0858c3d --- /dev/null +++ b/training/conf/experiment/vqvae_experiment.yaml @@ -0,0 +1,13 @@ +defaults: + - override /network: vqvae + - override /criterion: mse + - override /optimizer: madgrad + - override /lr_scheduler: one_cycle + - override /model: lit_vqvae + - override /dataset: iam_extended_paragraphs + - override /trainer: default + - override /callbacks: + - wandb + +load_checkpoint: null +logging: INFO -- cgit v1.2.3-70-g09d2