Begin implementing a training edge case to pickup where the training …

…left off if interrupeted.

train.py

@@ -397,6 +397,7 @@ def main():
         criterion,
         scheduler,
         tokenizer,
+        resume_from_checkpoint=resume_from_checkpoint,
     )
 
     # Finish logging and close the Weights & Biases run