Hello Sai_tharun,
I have implemented the above logic to return two output but i want i check point my model , in the current setup my model is a class and it doesn’t have the property of model.state_dict(), how can i implement the same.
Here is my code:
class resnet(nn.Module):
def init(self,original_model):
super().init()
self.original_model = original_model
self.features = nn.Sequential(*list(self.original_model.children())[:-2])
def forward(self,x):
    x = self.features(x)
    x = x.view(x.size(0),-1)
    vowel = self.original_model.fc(x)
    consonant = self.original_model.fc1(x)
    return vowel,consonant

but i want to save the check point as in when i get the least error while training, how can i implement the same.