我将 keras.applications.Xception 用于多标签分类任务。 我试图通过设置classifier_activation='sigmoid' 并使用二进制焦点损失设置损失函数来做到这一点。 对于初始训练,我将 weights=None 设置为随机统一权重初始化,但我想在会话之间保存并恢复训练。
问题是,当尝试使用 weights='best_weights.hdf5' 恢复训练时,出现错误 (https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/keras/applications/imagenet_utils.py#L430)
Only None and softmax activations are allowed for the classifier_activation argument when using pretrained weights, with include_top=True
。所以,我刚刚在第 427 行添加了 activations.get('softmax')
,问题似乎已经解决了。