Commit 62972be1 authored by Verena Praher's avatar Verena Praher

restore conv11, doesn't seem to work without it

parent 5ff2f93a
......@@ -118,7 +118,9 @@ class ModelMidlevel(BasePtlModel):
self._load_model(load_from, map_location, on_gpu)
if dataset == 'mtgjamendo':
self.fc_mtg1 = nn.Linear(512, 56)
self.fc_mtg1 = nn.Sequential(
# nn.AdaptiveAvgPool2d((1, 1)),
nn.Linear(256, 56))
# self.fc_mtg2 = nn.Linear(10, 56)
for name, param in self.named_parameters():
......@@ -145,9 +147,12 @@ class ModelMidlevel(BasePtlModel):
x = x.view(x.size(0), -1)
x = self.fc_ml(x)
if self.dataset=='mtgjamendo':
x = self.conv11(x)
# print(x.shape)
x = x.view(x.size(0), -1)
#x = self.fc_mtg1(x)
logit = nn.Sigmoid()(self.fc_mtg1(x))
# print(x.shape)
x = self.fc_mtg1(x)
logit = nn.Sigmoid()(x)
return logit
return x
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment