PyTorch version: 2.9.0+rocm6.4
Training completed successfully