随笔分类 - PyTorch
发表于 2021-08-29 15:25阅读：662评论：0推荐：0
摘要：前言：看过访问远程服务器tenserboard可视化方法，需要建立建立ssh隧道, 但是现在只需要加上--bind_all参数即可。 输入以下命令启动服务, logdir/ 目录下就是tenserboard的event存放目录, --bind_all, 即可让可视化被远程访问。 tensorboar
阅读全文 »
发表于 2021-07-18 22:27阅读：3428评论：0推荐：1
摘要：错误信息 类型错误, 计算loss值的函数传入的参数类型不统一。 解决方法 查看上文loss计算代码部分的参数类型，如loss=f.mse_loss(out,label)，检查out和label的类型都是torch.float类型即可。使用label.dtype查看tensor的类型。
阅读全文 »
发表于 2021-04-18 17:12阅读：131评论：0推荐：0
摘要：model.train() and model.eval() 定义了模型的训练和测试两种模式, 对部分层有影响, 如Dropout和BatchNormal Dropout: 为了增加模型的泛化能力(防止过拟合),训练过程中可以随机屏蔽掉一些神经元，测试时恢复。 BatchNormal: 训练过程中有
阅读全文 »
