PyTorch version: 2.11.0
Training completed successfully