![](https://img-blog.csdnimg.cn/20201014180756923.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
文章平均质量分 50
Chenguanxixixi
这个作者很懒,什么都没留下…
展开
-
pytorch1.60 torch.nn在pycharm中无法自动智能提示
转载自https://blog.csdn.net/winter2121/article/details/108799776问题描述安装了pytorch最新版本1.6之后,在pycharm中编辑python代码时,输入torch.nn.看不到提示了,比如torch.nn.MSELoss()。而在1.4及以前的版本中,直接输入torch.nn.就会自动提示出很多torch.nn.modules中的API。该问题的讨论在前几年有过不少(https://www.zhihu.com/question/27964转载 2022-01-19 20:14:03 · 644 阅读 · 1 评论 -
Pytorch的DataSet和DataLoader类编写套路
DataSet类的说明An abstract class representing a :class:Dataset.All datasets that represent a map from keys to data samples should subclassit. All subclasses should overwrite :meth:`__getitem__`, supporting fetching adata sample for a given key. Subclasses原创 2022-01-18 16:25:17 · 903 阅读 · 0 评论 -
卷积层和池化层后的新图片尺寸大小计算
由于pytorch没有自动计算卷积层和池化层后的图片尺寸,在全连接层Linear中需要自己计算图片尺寸,因此需要记住这几个公式。卷积层W:矩阵宽,H:矩阵高。F:卷积核宽和高P:padding(需要填充的0的个数N:卷积核的个数S:步长因此width=(W−F+2P)/S+1width = (W - F + 2P)/ S + 1width=(W−F+2P)/S+1height=(H−F+2P)/S+1height = (H - F + 2P) / S + 1height=(H−F+2P)/原创 2021-10-26 15:04:59 · 4570 阅读 · 3 评论