PyTorch大模型训练适配