minor update

RihabFekii · Apr 3, 2023 · 5fdf06e · 5fdf06e
1 parent d816c4a
commit 5fdf06e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dvc.yaml b/dvc.yaml
@@ -7,8 +7,9 @@ stages:
  - src/params.yaml
  - src/utils.py
  outs:
- - models/model.pt:
- persist: true
+ - models/
+ - reports/train_params.yaml:
+ cache: false
  metrics:
  - reports/train_metrics.csv:
  cache: false