s444409/ium_444409
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

DVC Fixes

Browse Source
This commit is contained in:
Marcin Kostrzewski 2022-06-05 22:19:45 +02:00
parent 3f7b2ec7cf
commit 114109ecf6
4 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dvc.lock Normal file
View File

@ -0,0 +1,4 @@
schema: '2.0'
stages:
prepare:
cmd: download_dataset.sh

5
dvc.yaml
View File

@ -1,6 +1,6 @@
stages: stages:
prepare: prepare:
cmd: ./download_dataset.sh cmd: download_dataset.sh
train: train:
cmd: python3 train_model.py cmd: python3 train_model.py
@ -11,4 +11,5 @@ stages:
plots: plots:
- trend.png - trend.png
params: params:
- model_params.yaml - batch_size
- epochs

3
model_params.yaml
View File

@ -1,3 +0,0 @@
train:
batch_size: 64
epochs: 5

2
params.yaml Normal file
View File

@ -0,0 +1,2 @@
batch_size: 64
epochs: 5