We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0252bda commit a1271e8Copy full SHA for a1271e8
siamese_network/main.py
@@ -27,7 +27,7 @@ class SiameseNetwork(nn.Module):
27
def __init__(self):
28
super(SiameseNetwork, self).__init__()
29
# get resnet model
30
- self.resnet = torchvision.models.resnet18(pretrained=False)
+ self.resnet = torchvision.models.resnet18(weights=None)
31
32
# over-write the first conv layer to be able to read MNIST images
33
# as resnet18 reads (3,x,x) where 3 is RGB channels
@@ -53,7 +53,7 @@ def __init__(self):
53
54
def init_weights(self, m):
55
if isinstance(m, nn.Linear):
56
- torch.nn.init.xavier_uniform(m.weight)
+ torch.nn.init.xavier_uniform_(m.weight)
57
m.bias.data.fill_(0.01)
58
59
def forward_once(self, x):
0 commit comments