27 lines
672 B
YAML
27 lines
672 B
YAML
defaults:
|
|
- default.yaml
|
|
|
|
watch_model:
|
|
_target_: src.callbacks.wandb_callbacks.WatchModel
|
|
log: "all"
|
|
log_freq: 100
|
|
|
|
upload_code_as_artifact:
|
|
_target_: src.callbacks.wandb_callbacks.UploadCodeAsArtifact
|
|
code_dir: ${work_dir}/src
|
|
|
|
upload_ckpts_as_artifact:
|
|
_target_: src.callbacks.wandb_callbacks.UploadCheckpointsAsArtifact
|
|
ckpt_dir: "checkpoints/"
|
|
upload_best_only: True
|
|
|
|
log_f1_precision_recall_heatmap:
|
|
_target_: src.callbacks.wandb_callbacks.LogF1PrecRecHeatmap
|
|
|
|
log_confusion_matrix:
|
|
_target_: src.callbacks.wandb_callbacks.LogConfusionMatrix
|
|
|
|
log_image_predictions:
|
|
_target_: src.callbacks.wandb_callbacks.LogImagePredictions
|
|
num_samples: 8
|