Train a small PyTorch model on one GPU for 90 minutes using account school_g001 on the GPU partition.
