/TargetDrop

TargetDrop: A Targeted Regularization Method for Convolutional Neural Networks

Primary LanguagePython

收录到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)

算法框架