Introduction

The PyTorch GPU2Ascend tool can migrate GPU-based training scripts into NPU-based scripts at a high speed, minimizing the workload of developers. This sample helps developers quickly experience the migration efficiency of the PyTorch GPU2Ascend tool. GPU-based training scripts can run on NPUs after being successfully migrated.

This sample uses the ResNet-50 model and the CIFAR-100 dataset.