[Keras] Keras使用中遇到的一些坑~

xiaoxiao2021-02-28  57

主要内容

如题,遇到的时候会黑人问号脸的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的倍数…

转载请注明原文地址: https://www.6miu.com/read-2631028.html

最新回复(0)