Hubinbin 的博客

Tensorflow2.x 模型保存和加载Tensorflow官网教程给出了两种保存整个模型的方法：SavedModel formatHDF5 format在自定义模型时遇到了一个问题：Traceback (most recent call last): File "/data/projects/sysml/resNetAll/main.py", line 100, in <module> 'learning_rate':learning_rate,'decay_rat

Tensorflow 2.1 自定义网络层的方法及其注意事项具体定义方法在tensorflow官网 tf2.1 API 中定义的很清楚，但其中有需要注意的事项，是官方API没有写清楚的定义方法官网给出的定义方式：class MyDenseLayer(tf.keras.layers.Layer): def __init__(self, num_outputs): super(MyDenseLayer, self).__init__() self.num_outputs = nu