解决办法:举例如下:
z_box=z_box.cpu().detach().numpy()
x_box = x_box.cpu().detach().numpy()
其中z_box和x_box都是放在cuda上的tensor,train训练时要将它们从tensor变成numpy,需要同时加上.cpu()和.detach()。
test测试或track时,由于不需要反向传播,就不需要加上detach(),举例如下:
pred_score=pred_score.cpu().numpy()
pred_reg=pred_reg.cpu().numpy()
希望能够帮助到你!加油ψ(`∇´)ψ(✿◡‿◡)[]~( ̄▽ ̄)~*