![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
黑曜石小刀
这个作者很懒,什么都没留下…
展开
-
pytorch centerloss代码注释
代码是网上找的,运行出来效果不错class CenterLoss(nn.Module): """Center loss. Reference: Wen et al. A Discriminative Feature Learning Approach for Deep Face Recognition. ECCV 2016. Args: num_classes (int): number of classes. feat_dim (in原创 2021-09-10 20:26:28 · 707 阅读 · 1 评论 -
pytorch简单项目模板
pytorch简单项目模板初学pytorch,但是不知道一个完整的项目应该怎么写,网上的模板大多数都很复杂看不懂,终于找到一个适合新手用的模板,包含数据生成器、模型定义、训练、模型保存、记录最佳模型、画图分析。模板地址:https://github.com/KinglittleQ/Pytorch-Templatepytorch项目流程一般来说,用pytorch编写深度学习模型,一般需要编写一下几个部分:模型定义数据处理和加载训练模型测试模型训练过程可视化这里先给出模板的总体,之后再详细说明原创 2021-07-30 17:37:04 · 874 阅读 · 0 评论 -
卷积、池化操作计算操作后的矩阵维度
N = (W − F + 2P )/S+1N : output_shape 为 N x NW : input_shape 为 W×WF : Filter 大小 F×FP : Padding 大小S : 步长 stride原创 2021-07-24 14:38:28 · 518 阅读 · 0 评论