Upload train_autogrow.py with huggingface_hub

train_autogrow.py

@@ -126,7 +126,7 @@ if __name__ == "__main__":
     api = HfApi()
     start_time = time.time()
     step = 0
-    while True:
+    while True: # UNLIMITED STEPS - Let him grow!
         for batch in dataloader:
             optimizer.zero_grad()
             input_ids = batch["input_ids"].to("cuda")