не может преобразовать тензор CUDA в NumPy. Использование Tensor.cpu (), чтобы скопировать тензор памяти хоста первым

NoSuchKey

рекомендация

отblog.csdn.net/weixin_42528089/article/details/104926787