深度学习笔记
zzm8421
.
展开
-
CrossEntroy的计算
categorical_crossentropy用于多分类。tensorflow2.0+中提供了计算方法:y_true = tf.constant([[0,1,0],[0,0,1]])# 这里的y_pred是经过softmax激活函数运算后的值y_pred = tf.constant([[0.05,0.95,0],[0.1,0.8,0.1]])# 如果y_pred不经过softmax激活函数,这里加上参数 from_logits=Truetf.keras.losses.categor原创 2021-07-23 14:25:21 · 146 阅读 · 0 评论 -
Tensorflow2 加载别人模型权重失败时的一种解决方法:每一层分别设置值
有时自己建立模型导入别人模型参数时会因为版本等各个问题导致导入失败,这时可以一层一层的给网络赋值权重。本人在加载某个模型的权重时原创 2021-06-01 17:08:25 · 1774 阅读 · 0 评论