PyTorch version: 2.9.1
Training completed successfully