search

speedinghzl / pytorch-segmentation-toolbox

PyTorch Implementations for DeeplabV3 and PSPNet
MIT License
772 stars 167 forks source link

Dropout2d on the PSPmodule #50

Open erichhhhho opened 5 years ago

erichhhhho commented 5 years ago

Hi, may I ask why you add nn.Dropout2d(0.1) in the PSP module, which is different from the original implementation