在使用torch-pruning给siamRPN++模型剪枝时,构建剪枝器的过程中报错如下:Parameter containing: tensor([ 0.2020, 0.3365, -0.5385], device='cuda:0', requires_grad=True)
一步一步调试进去,会进入model的forward函数,都能顺利通过,没有报错,但是调试出来回到构建剪枝器这里,依然报错。
有没有小伙伴知道这是怎么回事呢?求指导!感谢~
在使用torch-pruning给siamRPN++模型剪枝时,构建剪枝器的过程中报错如下:Parameter containing: tensor([ 0.2020, 0.3365, -0.5385], device='cuda:0', requires_grad=True)
一步一步调试进去,会进入model的forward函数,都能顺利通过,没有报错,但是调试出来回到构建剪枝器这里,依然报错。
有没有小伙伴知道这是怎么回事呢?求指导!感谢~