Commit 146bd8c1 authored by Verena Praher's avatar Verena Praher

allow finetuning off conv11

parent 39dc905f
......@@ -124,7 +124,8 @@ class ModelMidlevel(BasePtlModel):
# self.fc_mtg2 = nn.Linear(10, 56)
for name, param in self.named_parameters():
if 'mtg' in name:
if 'mtg' or 'conv11' in name:
print(name, param)
param.requires_grad = True
else:
param.requires_grad = False
......
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