PyTorch version: 2.11.0a0+git069d5d8
Training completed successfully