AntoineH/smart_augmentation
1
0
Fork 0
mirror of https://github.com/AntoineHX/smart_augmentation.git synced 2025-05-04 12:10:45 +02:00
Code Issues Projects Releases Packages Wiki Activity

Resultats avec early stop sur data test

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-11-14 13:13:03 -05:00
parent 93d91815f5
commit 67f99f3354
85 changed files with 42780 additions and 6607 deletions
Download patch file Download diff file Expand all files Collapse all files

3
higher/model.py
View file

@ -55,6 +55,7 @@ class LeNet(nn.Module):
#https://github.com/xternalz/WideResNet-pytorch/blob/master/wideresnet.py
#https://github.com/arcelien/pba/blob/master/pba/wrn.py
#https://github.com/szagoruyko/wide-residual-networks/blob/master/pytorch/resnet.py
class BasicBlock(nn.Module):
def __init__(self, in_planes, out_planes, stride, dropRate=0.0):
super(BasicBlock, self).__init__()
@ -145,4 +146,4 @@ class WideResNet(nn.Module):
return super(WideResNet, self).__str__()
def __str__(self):
return "WideResNet(s{}-d{})".format(self.kernel_size, self.depth)
return "WideResNet(s{}-d{})".format(self.kernel_size, self.depth)