摘要: 1.写到一个类别下 https://www.pythonheidong.com/blog/article/674750/850e5ee9aa39c1d29ff5/ from torch.utils.tensorboard import SummaryWriter import numpy as np 阅读全文
posted @ 2021-04-04 23:38 lypbendlf 阅读(5214) 评论(0) 推荐(0) 编辑
摘要: 1.作用 运行model.eval()后批归一化层和dropout层就不会在推断时有效果。如果没有做的话,就会产生不连续的推断结果。 2.model.eval()和with torch.no_grad() https://discuss.pytorch.org/t/model-eval-vs-wit 阅读全文
posted @ 2021-04-04 16:29 lypbendlf 阅读(3127) 评论(0) 推荐(0) 编辑
摘要: 转自:https://blog.csdn.net/qq_39938666/article/details/88809726 1.相同点 两者的公式都是这样的。 2.torch.nn.Sigmoid() 它是一个网络层的class类(首字母为大写的),需要先初始化一个网络层,然后forward输入进参 阅读全文
posted @ 2021-04-04 16:19 lypbendlf 阅读(2213) 评论(0) 推荐(0) 编辑