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