PyTorch中定义模型时,有时候会遇到self.register_buffer('name', Tensor)的...[作者空间]
本文首发自【简书】用户【西北小生_】的博客,转载请注明出处! PyTorch之HOOK——获取神经网络特征和梯度的...[作者空间]
之前写过一篇自动计算模型参数量、FLOPs、乘加数以及所需内存等数据[https://www.jianshu.co...[作者空间]
本文通过一个例子实验来观察并讲解PyTorch中model.modules(), model.named_modu...[作者空间]
前面写过一篇PyTorch保存模型的文章:Pytorch模型保存与加载,并在加载的模型基础上继续训练 ,简单介绍了...[作者空间]
1. torch.squeeze(tensor) 和numpy等库函数中的squeeze()函数作用一样,torc...[作者空间]