昇腾论坛

yolov9迁移到ascend910

已解决

新人帖

发表于2024-08-03 15:52:32

0 查看

在本地gpu上能跑通yolov9，在迁移到ascend时使用自动迁移加入了如下代码

import torch 
import torch_npu 
from torch_npu.contrib import transfer_to_npu

运行后显示梯度溢出是怎么回事啊

匿名回复

我要发帖子