作者:迷-糊-娃娃 | 来源:互联网 | 2023-05-18 04:00
主要内容如题,遇到的时候会黑人问号脸的KerasIssues1.ClassWeights问题问题描述:Keras在多分类或多二分类时使用ClassWeights时
主要内容
如题,遇到的时候会黑人问号脸的Keras Issues
1. Class Weights问题
问题描述:Keras在多分类或多二分类时使用Class Weights时会报错,提示class not exists,这是一个已知的Keras Bug,有一个work around的方法是,在label vector前面加两个位置,分别赋值0和1,举个例子:
# Original label vector
y_ori = [1, 0, 0, 0]
# Work Around label vector for Keras
y_ = [0, 1, 1, 0, 0, 0]
2. Multi-GPU Compile后遇到的问题
第一:神TM在Compile 到Multi-GPU之后,保存模型时使用save函数就报错了,需要使用save_weight
第二:使用Multi-GPU Compile 的模型在predict的时候,batch的size必须是GPU number的倍数…