Spaces:
Sleeping
Sleeping
File size: 852 Bytes
9e4713f a2d2c23 9e4713f a2d2c23 9e4713f 743cb9b 4c71213 9e4713f 4c71213 743cb9b 4c71213 743cb9b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
schema: '2.0'
stages:
train:
cmd: poetry run train
deps:
- path: data/raw/arxivData.json
md5: a314e2f4eab544a46e6f95802ecde647
size: 72422946
- path: shad_mlops_transformers/model.py
md5: 465d64e91ac7d713e57442c932a665ea
size: 2889
- path: shad_mlops_transformers/trainer.py
md5: 39d02f490890ef306b1165fcc29b8895
size: 7495
params:
shad_mlops_transformers/config.py:
Config.batch_size: 64
Config.epochs: 15
Config.random_seed: 42
Config.test_size: 0.2
outs:
- path: data/models/model.torch
md5: cad526c49e9a9ef312e820ba4cc93442
size: 438842393
- path: data/plots/train.png
md5: a23fefe43ee875860f875123bd402522
size: 40498
- path: data/raw/mapper.json
md5: e4c22adf587676d22ed58c812f41f409
size: 855
|