added .yaml

2023-05-13 17:16:16 +02:00 · 2023-05-13 17:16:16 +02:00 · 1ea45aca70
commit 1ea45aca70
parent bd3269f4b3
1 changed files with 11 additions and 0 deletions
--- a/dvc.yaml
+++ b/dvc.yaml
@ -0,0 +1,11 @@
+stages:
+  train:
+    cmd: python ./train.py 10
+    deps:
+    - Customers.csv
+    - train.py
+  test:
+    cmd: python ./test.py
+    deps:
+    - cifar_net.pth
+    - test.py