Skip to content

darkknightzh/DropBlock_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DropBlock_pytorch

This is my second time to upload code to github, if there is something wrong with README and LICENSE, please let me know. I will modify or delete the code or project if necessary.

other implementations of DropBlock:

Randl:DropBlock-pytorch

DHZS:tf-dropblock

An implementation of DropBlock in PyTorch. DropBlock is a replacement for dropout which zero-es entire spatial blocks instead of single pixels. For more information, check the paper: DropBlock: A regularization method for convolutional networks

Inspired by DHZS:tf-dropblock, I modified the code from Randl:DropBlock-pytorch, and use max pooling instead of conv to get the mask of DropBlock(named DropBlock2DV2 in the code). All the code except DropBlock2DV2 comes from Randl:DropBlock-pytorch, you can go to Randl:DropBlock-pytorch and (原)DropBlock A regularization method for convolutional networks for more information.

About

The PyTorch implementation of DropBlock

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages