feat(model): train on 15 epochs

dvc.lock

@@ -15,15 +15,16 @@ stages:
     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: 3a56baed312933280d01e30f86bd8b57
-      size: 440542446
+      md5: cad526c49e9a9ef312e820ba4cc93442
+      size: 438842393
     - path: data/plots/train.png
-      md5: 87c500a795ca97c14b84a6c0367d7b78
-      size: 44168
+      md5: a23fefe43ee875860f875123bd402522
+      size: 40498
     - path: data/raw/mapper.json
       md5: e4c22adf587676d22ed58c812f41f409
       size: 855

dvc.yaml

@@ -10,6 +10,7 @@ stages:
           - Config.batch_size
           - Config.random_seed
           - Config.test_size
+          - Config.epochs
     outs:
       # NOTE должно совпадать с конфигом
       - data/models/model.torch

shad_mlops_transformers/config.py

@@ -11,7 +11,7 @@ class Config(BaseSettings):
     checkpoints_folders = data_dir / "checkpoints"
     plots_dir: Path = data_dir / "plots"
     batch_size: int = 64
-    epochs: int = 100
+    epochs: int = 15
     random_seed: int = 42
     test_size: float = 0.2
     weights_path: Path = data_dir / "models" / "model.torch"