[tf]感觉keras就是个rubbish一个没有完工的半成品,bug遍地,以后能不用就不不用了

如果用上面注释中的keras代码,test的时候就加载不了dense_1/bias,用了下面的代码就正常了。真的rubbish

class Project_layer(object):
  def __init__(self,num_classes):
    # self.multiclass_dense_layer = K.layers.Dense(num_classes,activation='relu')  
    # super()
    self.linear_model = tf.layers.Dense(units=num_classes,activation='relu')

  def __call__(self,x):
    # return self.multiclass_dense_layer(x)
    # return subgraph(x)
    return self.linear_model(x)
    # return tf.layers.dense(x,self.num_classes,activation='relu')
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容