Size of checkpoint.pth file

Hi everyone :slight_smile:

I have two CNNs out of which I save a checkpoint each, as a .pth file. The first model has a regular forward method whereas the second one has a forward method where I loop over each channel in an RGB image to process them separately.

I noticed that the .pth file from the CNN with a loop is roughly 3x smaller compared to the regular one but both models have the same number of parameters… Can anyone explain this to me?

Any help is very much appreciated!

All the best
snowe