如何为子类模型实现get_config方法?

时间:2019-07-03 05:31:27

标签: tensorflow keras

在TF 2.0中,我建立了一个模型,我试图将其作为一个整体保存,但是失败了(因为根据TF文档它无法序列化)。取而代之的是,我节省了重量,并且运行良好,但是我也想节省架构。 get_config方法引发错误,表明未实现。我试图将其转换为json,但归结为相同的错误。我如何自己实现它?

我注意到get_config适用于组成子类NN的各个层,但不适用于整个NN。

0 个答案:

没有答案