https://lukemelas.github.io/image-colorization.html 在以下代码中，同时在各个时期进行测试和训练，我希望在几个这样的时期之后/在对一组图像进行训练之后创建一个.h5模型文件。请帮助我将代码的哪一部分放在哪里？谢谢。
我尝试使用h5py创建.h5文件，发布它也创建了h5文件，但是当我将其输入模型时，它没有用，而是出现了一些文件截断问题。也许我不知道将h5文件写在哪里。
链接中提供的整个代码。
我希望创建一个.h5文件，该文件输入我的模型后不会出现问题。谢谢