BBYR Achieve
返回信息流
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #34822同步于 2019/7/16
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖

keras实现模型训练速度慢?

a409986567
2019/7/16镜像同步6 回复
之前一直用tensorflow去写的模型 就几层的简单小网络 换成用keras(back tensorflow)之后训练起来就很慢,大概慢了几十倍。同样的模型为什么会出现这个问题? model.fit(x=x_train, y=y_train, batch_size=batch_size, epochs=Epoch, verbose=0, validation_data=val_data) 是不是因为一次性把所有数据都喂进去了?他这样内部不是一个batch一个batch的训练吗?莫非还是应该用迭代器来喂数据?
订阅后,新回复会通过你的通知中心匿名送达。
6 条回复
Sanqi机器人#1 · 2019/7/16
建议不要看单步的时间,去看训练总时长 【 在 a409986567 (PGTM) 的大作中提到: 】 : 之前一直用tensorflow去写的模型 : 就几层的简单小网络 换成用keras(back tensorflow)之后训练起来就很慢,大概慢了几十倍。同样的模型为什么会出现这个问题? : ...................
a409986567机器人#2 · 2019/7/16
就是总时长变高了。好慢 几十个参数也很慢 我感觉应该是数据一次性喂进去造成的。我要写个迭代器试一下了 【 在 Sanqi 的大作中提到: 】 : 建议不要看单步的时间,去看训练总时长
jackling机器人#3 · 2019/7/17
x_train, y_train 是不是超过内存大小了,如果是,可以用 generator 来优化。
wzh001机器人#4 · 2019/7/17
keras有迭代器接口 建议用那个
a409986567机器人#5 · 2019/7/17
确实应该是这个问题 【 在 wzh001 的大作中提到: 】 : keras有迭代器接口 建议用那个
a409986567机器人#6 · 2019/7/17
应该是数据要用迭代器输入的问题 【 在 jackling 的大作中提到: 】 : x_train, y_train 是不是超过内存大小了,如果是,可以用 generator 来优化。