Keras
Regnaiq
只是一个学生, 学无止境. 慢慢来, 比较快.
展开
-
Keras指定model.fit()的输出
当verbose=1的时候会打印出所有指标和loss, 在多输出的情况下更是一团乱麻.函数, 然后利用callbacks调用即可.原创 2023-08-08 10:36:01 · 688 阅读 · 0 评论 -
Keras: InternalError: Blas xGEMM launch failed
原因是GPU内存不够了.解决方法有以下几种:修改cuda或者cuDnn的版本. 但是好麻烦. 详细版本见这里.Reference[1] Blas xGEMMBatched launch failed的出现原因原创 2022-06-30 14:44:20 · 1078 阅读 · 0 评论 -
Keras如何应对无法一次性读入内存的大文件?
对于很小的数据集来说, 我们可以直接读进内存来训练模型:model.fit(x_train, y_train)那么如果数据量太大, 比如几十G, 该怎么办呢?什么是generator好在Keras的fit函数支持了generator. 我们可以分batch读取文件来进行训练, 达到节省内存的目的.网上很多介绍generator的文章都差不多过时了, 推荐python wiki.简单来说, 就是可以定义一个函数, 每次调用返回不会停止程序, 而是返回中间结果, 下次调用的时候会继续运行循环. 比原创 2021-08-02 11:23:22 · 594 阅读 · 0 评论