:rocket: Finetuned for 18hrs on my laptop

app.py

@@ -245,8 +245,8 @@ def start():
                                                                         trust_remote_code=True)
   args = {
     "output_dir": "output_hyena_dna-mqtl_classification",
-    "num_train_epochs": 10,
-    "max_steps": 40_000,  # train 36k + val 4k = 40k
+    "num_train_epochs": 1,  # I don't get it, it ain't working :/ just set it to 1
+    "max_steps": 2_0000,  # my small dataset -> train 18k + val 2k = 20k. will take 17h to train :'(
     # Set the number of steps you expect to train, originally 1000, takes too much time. So I set it to 10 to run faster and check my code/pipeline
     "run_name": "laptop_run_hyena_dna-mqtl_classification",  # Override run_name here
     "per_device_train_batch_size": 1,