• 博客园logo
  • 会员
  • 众包
  • 新闻
  • 博问
  • 闪存
  • 赞助商
  • HarmonyOS
  • Chat2DB
    • 搜索
      所有博客
    • 搜索
      当前博客
  • 写随笔 我的博客 短消息 简洁模式
    用户头像
    我的博客 我的园子 账号设置 会员中心 简洁模式 ... 退出登录
    注册 登录

LR233

  • 博客园
  • 联系
  • 订阅
  • 管理

公告

View Post

4、model.fit_generator()

1、model.fit_generator()

    首先,利用keras,搭建顺序模型,完成搭建后,我们需要将数据送入模型进行训练,送入数据的方式有很多种,models.fit_generator()是其中一种方式。

  具体说,model.fit_generator()是利用生成器,分批次向模型送入数据的方式,可以有效节省单次内存的消耗。

  利用Python的生成器,逐个生成数据的batch并进行训练。生成器与模型将并行执行以提高效率。

  例如,该函数允许我们在CPU上进行实时的数据提升,同时在GPU上进行模型训练。

2、参数

1、generator:生成器函数
2、validation_data:生成验证集的生成器
3、epochs:整数,数据迭代的轮数
4、steps_per_epoch:整数,当生成器返回steps_per_epoch次数据时计一个epoch结束,执行下一个epoch。一般设置为:len(image_files) // batch_size
5、validation_steps: 当validation_data为生成器时,本参数指定验证集的生成器返回次数。一般设置为:len(valid_image_files) // batch_size

3、生成器generator

参考:生成器 - 廖雪峰的官方网站 (liaoxuefeng.com)

在Keras中导入测试数据的方法 - 简书 (jianshu.com)

3.1、小括号

3.2、边迭代边输出

 

 

posted on 2022-10-24 21:51  LR233  阅读(558)  评论(0)    收藏  举报

刷新页面返回顶部
 
博客园  ©  2004-2025
浙公网安备 33010602011771号 浙ICP备2021040463号-3