I have a very simple model with 2 conv and 2 linear layers.
the input and output of the linear layers depends on the batch size.
I am having problem(size mismatch) when I try to save the model weights with a batch size and restore with another batch size.
How can I make save/restore independent from the batch size? Can you point me to a sample model?
Thanks!