返回顶部
摘要: for inputs, labels in train_loader: # 使用numpy的transpose函数调整维度顺序 inputs = np.transpose(inputs, (0, 3, 1, 2)) #将原输入数据最后一个维度换到第二个维度 inputs = inputs.to(de 阅读全文
posted @ 2023-09-08 23:02 欣杰科技 阅读(220) 评论(0) 推荐(0)
摘要: 报错如下: RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Float' 一般来说这个问题是计算Loss时的报错。 解决方法: 将如下代码 loss_func(torch.squeez 阅读全文
posted @ 2023-09-08 22:40 欣杰科技 阅读(2433) 评论(0) 推荐(0)
摘要: 安装命令 pip install gymnasium[all] 如遇如下报错 error: command 'swig.exe' failed: No such file or directory [end of output] note: This error originates from a 阅读全文
posted @ 2023-09-08 11:47 欣杰科技 阅读(1241) 评论(0) 推荐(0)