Why pytorch's model is much smaller than torch's?

1

Hi,

I’m new to pytorch, and I find that pytorch’s model files size is much smaller than torch’s. For example: resent-18

pytorch’s model is 44.7MB, torch’s model is 138MB

Why there is so much difference ?

Thanks in advance.

2

torch’s models still saves gradWeight/gradBias i think. So pytorch saves half size on that.

1 Like
3

Yes， that’s right, after I set the gradWeight/gradBias to nil, the two model are nearly same.

Thanks!