No real checkpoint stored

[Y-Robin]

Hey together,

I recently came across your repository and found it interesting. However, if I try to follow the examples given, I come across some issues. The most problematic is that the model needs to be stored correctly. For some reason, my ckpt file only has 1kb, which seems not sufficient, especially if I try to test it. If I test it, I only get random noise in different colors. Can someone help me out?