标准的使用技巧.详细介绍可以参考《深度学习之TensorFlow:工程化项目实战》一书第6章
1.3 坑1 :keras自定义模型的默认输入
如果在TF1.x中代码第1.2小节第7行会有问题,它是一个函数不能充当一个层...=5, batch_size=batch_size)
再次运行,即可通过.
3 张量损失封装成损失函数
其实2节所介绍的方法,也可以再次封装成损失函数来进行执行.具体做法如下.
3.1 将张量损失封装成函数...vae_loss.同时又建立一个损失函数,在输入损失时,将模型的输出透传出来即可....)就是这种.2节和3节提供了2个这种模型的训练方法,都可以使用.
4 使用动态图训练
前面的1,2,3节都是使用keras的方式来训练模型.这种方法看是方便,但不适合模型的调试环节.尤其当训练种出现了None...具体例子,见配套的源码文件
5 以类的方式封装模型损失函数
为了代码工整,还可以将模型的整个过程封装起来,直接输出损失函数.
5.1 封装损失函数
将整个流程封装起来,具体如下:
class VAE(