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 faster filtering and convolution, but not working yet for BU

Browse source
This commit is contained in:
Marco Pedersoli 2020-06-13 20:47:19 -04:00
parent 9d68bc30bd
commit f7436d0002
5 changed files with 295 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/myresnet3.py
View file

@ -119,7 +119,7 @@ class ResNet(nn.Module):
def forward(self, x , stoch = False):
#if self.training==False:
# stoch=False
print(stoch)
#print(stoch)
# self.layer1.stoch=stoch
# self.layer2.stoch=stoch
# self.layer3.stoch=stoch