PyTorch version: 2.13.0a0+git976d8bf
Training completed successfully