热门标签 | HotTags
当前位置:  开发笔记 > 前端 > 正文

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

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


推荐阅读
author-avatar
迷-糊-娃娃
这个家伙很懒,什么也没留下!
PHP1.CN | 中国最专业的PHP中文社区 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved | 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有