Merge pull request #8 from yuriihavrylko/feature/l5t1-training

L5T1 Train model and log to W&B
yuriihavrylko · Feb 11, 2024 · 6a4e77d · 6a4e77d
2 parents c771641 + f73fe8c
commit 6a4e77d
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/app/requirements-dev.txt b/app/requirements-dev.txt
@@ -0,0 +1,3 @@
+ipykernel==6.28.0
+wandb==0.16.1
+evaluate==0.4.1
diff --git a/app/requirements.txt b/app/requirements.txt
@@ -1,2 +1,3 @@
 flask==3.0.0
 gunicorn==21.2.0
+transformers==4.36.2