YOLO run setup

colab_test_file.txt

@@ -1 +0,0 @@
-This file was created in Google Colab and pushed to the Hugging Face repo.

scripts/train.py

@@ -73,7 +73,7 @@ def main():
         per_device_train_batch_size=2,
         gradient_accumulation_steps=8,
         learning_rate=1e-4,
-        num_train_epochs=0.001,
+        num_train_epochs=3,
         logging_dir="./logs",
         logging_steps=10,
         save_steps=1000,