PyTorch version: 2.10.0
Training completed successfully