diff options
Diffstat (limited to 'training/conf/callbacks/wandb')
| -rw-r--r-- | training/conf/callbacks/wandb/captions.yaml | 4 | ||||
| -rw-r--r-- | training/conf/callbacks/wandb/checkpoints.yaml | 2 | ||||
| -rw-r--r-- | training/conf/callbacks/wandb/config.yaml | 2 | ||||
| -rw-r--r-- | training/conf/callbacks/wandb/predictions.yaml | 3 | ||||
| -rw-r--r-- | training/conf/callbacks/wandb/watch.yaml | 2 |
5 files changed, 7 insertions, 6 deletions
diff --git a/training/conf/callbacks/wandb/captions.yaml b/training/conf/callbacks/wandb/captions.yaml new file mode 100644 index 0000000..3215a90 --- /dev/null +++ b/training/conf/callbacks/wandb/captions.yaml @@ -0,0 +1,4 @@ +log_text_predictions: + _target_: callbacks.wandb.ImageToCaptionLogger + num_samples: 8 + on_train: true diff --git a/training/conf/callbacks/wandb/checkpoints.yaml b/training/conf/callbacks/wandb/checkpoints.yaml index a4a16ff..b9a3fd7 100644 --- a/training/conf/callbacks/wandb/checkpoints.yaml +++ b/training/conf/callbacks/wandb/checkpoints.yaml @@ -1,4 +1,4 @@ upload_ckpts_as_artifact: - _target_: callbacks.wandb_callbacks.UploadCheckpointsAsArtifact + _target_: callbacks.wandb.UploadCheckpointsAsArtifact ckpt_dir: checkpoints/ upload_best_only: true diff --git a/training/conf/callbacks/wandb/config.yaml b/training/conf/callbacks/wandb/config.yaml index 747a7c6..d841e94 100644 --- a/training/conf/callbacks/wandb/config.yaml +++ b/training/conf/callbacks/wandb/config.yaml @@ -1,2 +1,2 @@ upload_code_as_artifact: - _target_: callbacks.wandb_callbacks.UploadConfigAsArtifact + _target_: callbacks.wandb.UploadConfigAsArtifact diff --git a/training/conf/callbacks/wandb/predictions.yaml b/training/conf/callbacks/wandb/predictions.yaml deleted file mode 100644 index 573fa96..0000000 --- a/training/conf/callbacks/wandb/predictions.yaml +++ /dev/null @@ -1,3 +0,0 @@ -log_text_predictions: - _target_: callbacks.wandb_callbacks.LogTextPredictions - num_samples: 8 diff --git a/training/conf/callbacks/wandb/watch.yaml b/training/conf/callbacks/wandb/watch.yaml index 660ae47..1f60978 100644 --- a/training/conf/callbacks/wandb/watch.yaml +++ b/training/conf/callbacks/wandb/watch.yaml @@ -1,5 +1,5 @@ watch_model: - _target_: callbacks.wandb_callbacks.WatchModel + _target_: callbacks.wandb.WatchModel log_params: gradients log_freq: 100 log_graph: true |