Closed cumtwb closed 5 years ago
in my case one single ckpt occupies ~1.5G, but I set --light parameter to true because of poor GPU memory.
in my case one single ckpt occupies ~1.6G, but I set --light parameter to true because of poor GPU memory.
Using the light parameter my checkpoints were about 1.5G each.
Thank you! But I still have two problem:
light | false | true |
---|---|---|
x | 64 64 256 | 256 |
fc | 64 64 256 * 256 | 256 * 64 |
A float32 parameter occupies 4B, all parameters of one full fc layer occupy 64 64 256 256 4B = 1G. Therefore, the full a2b and b2a generators should occupy ~2G than the light generators. But my full ckpt occupies ~7.5G, is 6G than your light ckpt. Why? @heart4lor @HLearning @thewaifuai
Original checkpoint(not light model) size is 7.5G
I trained the network on my own dataset. But the size of saved checkpoint is 7.5G. Would you like to tell how much space your checkpoint occupies?