Deep Compression
文章平均质量分 66
Deep Compression
graph builder
生活点滴,细嚼、慢咽
展开
-
PyTorch 卷积与BatchNorm的融合(PyTorch官方)
前言:目录1 融合原理2 代码1 融合原理卷积的工作:BN的工作:带入的话可以推出来,融合后的新卷积:新的卷积就直接顺路完成BN的工作。2 代码import torchimport torch.nn as nnfrom utils.modules import DummyModuledef fuse(conv, bn): w = conv.weight mean = bn.running_mean var_sqrt = torch.sqrt.原创 2021-01-14 10:06:09 · 1815 阅读 · 0 评论 -
PyTorch 卷积与BatchNorm的融合
PyTorch 卷积与BatchNorm的融合 _Captain Jack原创 2019-10-21 10:57:06 · 857 阅读 · 0 评论