2024年4月11日
【Pytorch】ResNet50中替换BN层为IN层实现
摘要： 替换BN层为IN层 最近在做实验时，考虑将官方torchvision包中的Resnet模型进行一些更改，ResNet类中有个可选参数_norm_layer可以直接传入nn.InstanceNorm2d，默认为nn.BatchNorm，但是这样更改后，在使用官方的预训练权重时，会发生一些报错，BN层里 阅读全文
posted @ 2024-04-11 16:25 chendsome 阅读(104) 评论(0) 推荐(0)