logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

sam模型迁移昇腾训练loss不收敛问题的解决办法

npu在这里丢了image_embedding的梯度,因为该模型用到了Reg_op(RepeatInterleaveGrad)这个算子,接下来分析cann包是否有这个算子 发现是有的 但是通过进入python后调用torch_npu.repeat_interleave_backward_tensor这个函数,发现调用失败。功能说明:停止精度数据采集,在 start 函数之后的任意位置添加,若需要

文章图片
#迁移学习#pytorch
到底了