49 lines
1.0 KiB
Plaintext
49 lines
1.0 KiB
Plaintext
schema: '2.0'
|
|
stages:
|
|
prepare:
|
|
cmd: python prepare.py
|
|
deps:
|
|
- path: prepare.py
|
|
hash: md5
|
|
md5: a1c07d1d5caf6e5288560a189415785c
|
|
size: 2979
|
|
params:
|
|
params.yaml:
|
|
prepare:
|
|
data_dir: data/raw
|
|
save_dir: data/processed
|
|
train_valid_split:
|
|
- 0.7
|
|
- 0.3
|
|
random_seed: 0
|
|
outs:
|
|
- path: data/processed
|
|
hash: md5
|
|
md5: f4bf62ffa725ca9144b7852a283dc1da.dir
|
|
size: 295118798
|
|
nfiles: 60000
|
|
train:
|
|
cmd: python train.py
|
|
deps:
|
|
- path: data/processed
|
|
hash: md5
|
|
md5: f4bf62ffa725ca9144b7852a283dc1da.dir
|
|
size: 295118798
|
|
nfiles: 60000
|
|
- path: train.py
|
|
hash: md5
|
|
md5: b797ccf2fe61952bbf6d83fa51b0b11f
|
|
size: 3407
|
|
params:
|
|
params.yaml:
|
|
train:
|
|
data_dir: data/processed
|
|
epochs: 5
|
|
batch_size: 256
|
|
learning_rate: 5e-05
|
|
outs:
|
|
- path: model.pt
|
|
hash: md5
|
|
md5: 8ead2a7cd52d70b359d3cdc3df5e43e3
|
|
size: 102592994
|