我正在学习一个课程,它使用了一个不推荐使用的PyTorch版本,该版本不会根据需要将torch.int64更改为torch.LongTensor。抛出错误的当前代码部分是: loss = loss_fn(Ypred, Ytrain_) # calc loss on the prediction 不过,我认为应该在此部分更改dtype: Ytrain_ = torch.from_numpy当使用Ytrain_.dtype测试数据类型时,它返回torch.int64。我试图通过应用如下的long()函数来转换
我正试图通过ONNX (opset 11)用torch.nn.functional.grid_sample将模型从Pytorch (1.6)转换为TensorRT (7)。在Pytorch中运行时,定制的alternative I found ()非常慢,并且在将主循环转换为TRT时遇到了问题。我自己的双线性采样的实现(不仅仅是grid_sample,而是基于grid_sample的整个原始采样)在Pytorch中的执行速度要快得多,并成功地转换为TRT。(Nt, device=image.device) #tor