各位大佬,新人求助!
tensorflow吧
全部回复
仅看楼主
level 2
coolmaxguy 楼主
我用keras训练cnn的时候(后端是tf),刚开始跑第一个epoch就遇到了这样的提示信息
这是一个错误还是警告?因为我想着总共就几个G的数据,反正还有虚拟内存可用,这种程度应该没问题。然而这个信息出来以后就没变化了,这是卡住了还是处在漫长的训练中?请各位大佬指教!
2020年04月13日 03点04分 1
level 4
你这个明显是内存爆了
2020年04月13日 04点04分 2
level 4
你要一个batch一个batch的放,不要一次把全部数据喂进去。
2020年04月13日 04点04分 3
那么我写一个循环,每次循环喂一个batch,用fit函数训练一次,,fit函数可以多次调用的吧?
2020年04月14日 01点04分
@coolmaxguy 最好把你的数据用kears.data.Dataset封装一下吧,可以网上找一下教程
2020年04月15日 06点04分
level 1
我超10%,然后接着跑下去也可以
2020年04月14日 20点04分 4
1