Spaces:
Sleeping
Sleeping
File size: 852 Bytes
9e4713f a2d2c23 9e4713f 03dd850 9e4713f 743cb9b 4c71213 9e4713f 2b76117 03dd850 743cb9b 2b76117 743cb9b 03dd850 |
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: 6f76b90e32a94bea31bc9365494aceb9
size: 7503
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: 52fe46160071315c7bd7c7de66e6b59e
size: 438914457
- path: data/plots/train.png
md5: 32d0562b97cfc18e7b40836785f46aba
size: 45296
- path: data/raw/mapper.json
md5: 47758bce6ee09369e716fd8658ce3e01
size: 930
|