wandb.yaml 672 B

1234567891011121314151617181920212223242526
  1. defaults:
  2. - default.yaml
  3. watch_model:
  4. _target_: src.callbacks.wandb_callbacks.WatchModel
  5. log: "all"
  6. log_freq: 100
  7. upload_code_as_artifact:
  8. _target_: src.callbacks.wandb_callbacks.UploadCodeAsArtifact
  9. code_dir: ${work_dir}/src
  10. upload_ckpts_as_artifact:
  11. _target_: src.callbacks.wandb_callbacks.UploadCheckpointsAsArtifact
  12. ckpt_dir: "checkpoints/"
  13. upload_best_only: True
  14. log_f1_precision_recall_heatmap:
  15. _target_: src.callbacks.wandb_callbacks.LogF1PrecRecHeatmap
  16. log_confusion_matrix:
  17. _target_: src.callbacks.wandb_callbacks.LogConfusionMatrix
  18. log_image_predictions:
  19. _target_: src.callbacks.wandb_callbacks.LogImagePredictions
  20. num_samples: 8