可以快速上手的开发文档
如下所示：importioimporttorchimporttorch.onnxfrommodels.C3AEModelimportPlainC3AENetCBAMdevice=torch.device("cuda:0"..