AntoineH/BU_Stoch_pool
1
0
Fork 0
mirror of https://github.com/AntoineHX/BU_Stoch_pool.git synced 2025-05-04 09:40:46 +02:00
Code Issues Projects Releases Packages Wiki Activity

added forward fast in stoch.py, but it works only with ceil_mode=False

Browse source
This commit is contained in:
Marco Pedersoli 2020-06-18 12:27:45 -04:00
parent f7436d0002
commit 286921f8a0
3 changed files with 197 additions and 59 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/myresnet4.py
View file

@ -26,7 +26,7 @@ class SAvg_Pool2d(nn.Module):
out = savg_pool2d(x, self.stride, mode = self.mode,ceil_mode = self.ceil_mode)
return out
stochmode = 'sim'#'sim'#'stride''stoch'''
stochmode = 'stoch'#'sim'#'stride''stoch'''
finalstochpool = True
simmode = 'sbc'