unet训练自己的数据集（pytorch）

[free9318]

出现以下问题，请up主帮助解决一下
[{
"resource": "/d:/pythorch/unet-pytorch-main/train.py",
"owner": "python",
"code": {
"value": "reportMissingImports",
"target": {
"$mid": 1,
"path": "/microsoft/pyright/blob/main/docs/configuration.md",
"scheme": "https",
"authority": "github.com",
"fragment": "reportMissingImports"
}
},
"severity": 4,
"message": "无法解析导入“torch.cuda.amp”",
"source": "Pylance",
"startLineNumber": 326,
"startColumn": 14,
"endLineNumber": 326,
"endColumn": 28
}]