feat(model): train more

dvc.lock

@@ -7,11 +7,11 @@ stages:
       md5: a314e2f4eab544a46e6f95802ecde647
       size: 72422946
     - path: shad_mlops_transformers/model.py
-      md5: fb30feb5d875b7d2f25275b80c902d46
-      size: 2855
+      md5: 465d64e91ac7d713e57442c932a665ea
+      size: 2889
     - path: shad_mlops_transformers/trainer.py
-      md5: 89ba94c0a48741a998d08f379920a1e0
-      size: 7415
+      md5: 39d02f490890ef306b1165fcc29b8895
+      size: 7495
     params:
       shad_mlops_transformers/config.py:
         Config.batch_size: 64
@@ -19,11 +19,11 @@ stages:
         Config.test_size: 0.2
     outs:
     - path: data/models/model.torch
-      md5: b08f4441d6ad9042ebfa19035d6c365a
-      size: 438842393
+      md5: 3a56baed312933280d01e30f86bd8b57
+      size: 440542446
     - path: data/plots/train.png
-      md5: 627358cecadf27fb8a938e40b13bd993
-      size: 27309
+      md5: 87c500a795ca97c14b84a6c0367d7b78
+      size: 44168
     - path: data/raw/mapper.json
       md5: e4c22adf587676d22ed58c812f41f409
       size: 855

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 = 1
+    epochs: int = 100
     random_seed: int = 42
     test_size: float = 0.2
     weights_path: Path = data_dir / "models" / "model.torch"