bobo0810 / TargetDrop

TargetDrop: A Targeted Regularization Method for Convolutional Neural Networks

TargetDrop

收录到PytorchNetHub

说明

• 非官方
• 基于注意力机制的Drop正则化方法

环境

python版本	pytorch版本	系统
3.6	1.6.0	Ubuntu

使用

import TargetDrop
class MyNet():
    def __init__(self):
        ...
        self.target_drop=TargetDrop(in_channels)
    def forward(self, x):
        ...
        x=self.target_drop(x)

算法框架