When I tried to build a learner with other arch like vgg16, I’ve got an error but for resnet34 seems to work fine. Does anyone have the same experience?
champs.jaideep (jaideep v) #2
May be u should get useful idea from this
body = create_body(arch,True,-1)
h = create_head(num_features_model(body) * 2,256,,0.5,False)
self.cnn= nn.Sequential( body,h)
self.head = nn.Linear(num_features_model(self.cnn), 1)