I want to save the transfer-learned inception V3 model. I couldn’t able to figure out what to be saved apart from state_dict. I didn’t made any changes to convolution layers part by freezing them. I changed fully connected layer part by adding some layers and scaling to smaller number of outputs.
Now, what are the things to be stored in order to successfully load and use the model after saving it.
precisely
i want to fill these blanks
checkpoint = {
‘input_isize’: _________,
‘output_size’:102,
‘hidden_layers’: ________,
‘state_dict’: model.state_dict()}