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

主要内容

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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值