pytorch_mini_template A minimal template script for PyTorch TODO Checkpointing Rewrite into classes instead plain script