how can i keep part of my model ?
model = torchfcn.models.FCN32s(n_class=21)
checkpoint = torch.load("./logs/MODEL-fcn32s/checkpoint.pth.tar")
model.load_state_dict(checkpoint['model_state_dict'])
model = model[0:10] # to keep until 10th layer

What about the forward pass ? Do I have to redefine it ?