From 7268035fb9e57342612a8cc50a1fe04e8841ca2f Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Fri, 30 Jul 2021 23:15:03 +0200
Subject: attr bug fix, properly loading network

---
 training/conf/trainer/default.yaml | 27 +++++++++++----------------
 1 file changed, 11 insertions(+), 16 deletions(-)

(limited to 'training/conf/trainer')

diff --git a/training/conf/trainer/default.yaml b/training/conf/trainer/default.yaml
index 5797741..5ed6552 100644
--- a/training/conf/trainer/default.yaml
+++ b/training/conf/trainer/default.yaml
@@ -1,16 +1,11 @@
-seed: 4711
-wandb: false
-tune: false
-train: true
-test: true
-args:
-  stochastic_weight_avg: false
-  auto_scale_batch_size: binsearch
-  auto_lr_find: false
-  gradient_clip_val: 0
-  fast_dev_run: false
-  gpus: 1
-  precision: 16
-  max_epochs: 64
-  terminate_on_nan: true
-  weights_summary: top
+_target_: pytorch_lightning.Trainer
+stochastic_weight_avg: false
+auto_scale_batch_size: binsearch
+auto_lr_find: false
+gradient_clip_val: 0
+fast_dev_run: false
+gpus: 1
+precision: 16
+max_epochs: 64
+terminate_on_nan: true
+weights_summary: top
-- 
cgit v1.2.3-70-g09d2