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/callbacks/checkpoint.yaml | 12 ++++++------ training/conf/callbacks/early_stopping.yaml | 4 ++-- training/conf/callbacks/wandb.yaml | 6 +++--- 3 files changed, 11 insertions(+), 11 deletions(-) (limited to 'training/conf/callbacks') diff --git a/training/conf/callbacks/checkpoint.yaml b/training/conf/callbacks/checkpoint.yaml index 9216715..db34cb1 100644 --- a/training/conf/callbacks/checkpoint.yaml +++ b/training/conf/callbacks/checkpoint.yaml @@ -1,9 +1,9 @@ model_checkpoint: _target_: pytorch_lightning.callbacks.ModelCheckpoint - monitor: "val/loss" # name of the logged metric which determines when model is improving + monitor: val/loss # name of the logged metric which determines when model is improving save_top_k: 1 # save k best models (determined by above metric) - save_last: True # additionaly always save model from last epoch - mode: "min" # can be "max" or "min" - verbose: False - dirpath: "checkpoints/" - filename: "{epoch:02d}" + save_last: true # additionaly always save model from last epoch + mode: min # can be "max" or "min" + verbose: false + dirpath: checkpoints/ + filename: {epoch:02d} diff --git a/training/conf/callbacks/early_stopping.yaml b/training/conf/callbacks/early_stopping.yaml index 4cd5aa1..a188df3 100644 --- a/training/conf/callbacks/early_stopping.yaml +++ b/training/conf/callbacks/early_stopping.yaml @@ -1,6 +1,6 @@ early_stopping: _target_: pytorch_lightning.callbacks.EarlyStopping - monitor: "val/loss" # name of the logged metric which determines when model is improving + monitor: val/loss # name of the logged metric which determines when model is improving patience: 16 # how many epochs of not improving until training stops - mode: "min" # can be "max" or "min" + mode: min # can be "max" or "min" min_delta: 0 # minimum change in the monitored metric needed to qualify as an improvement diff --git a/training/conf/callbacks/wandb.yaml b/training/conf/callbacks/wandb.yaml index 6eedb71..0017e11 100644 --- a/training/conf/callbacks/wandb.yaml +++ b/training/conf/callbacks/wandb.yaml @@ -3,7 +3,7 @@ defaults: watch_model: _target_: callbacks.wandb_callbacks.WatchModel - log: "all" + log: all log_freq: 100 upload_code_as_artifact: @@ -12,8 +12,8 @@ upload_code_as_artifact: upload_ckpts_as_artifact: _target_: callbacks.wandb_callbacks.UploadCheckpointsAsArtifact - ckpt_dir: "checkpoints/" - upload_best_only: True + ckpt_dir: checkpoints/ + upload_best_only: true log_text_predictions: _target_: callbacks.wandb_callbacks.LogTextPredictions -- cgit v1.2.3-70-g09d2