基于TensorFlow的深度学习模型如何Debug

TensorFlow是谷歌2015年开源的用于深度学习算法开发的一套目前非常流行的框架,不仅在学术界而且在工业界都非常收欢迎。用TensorFlow实现一个神经网络十分的简洁,但是往往有时候你自己实现出来的神经网络的表现效果和预期有差别,或者说模型不工作。那么如何进行Debug,我自己最近有了一些小的体会,分享给大家。

本文主要分为两个部分,一部分是使用sess.run()方法获得模型运行期间的各个tensor的值,另一部分是使用TensorFlow自带的Debug工具tfdbg。

1.获取模型中tensor的值

在tensorflow构建的模型中,数据是以tensor的形式的存储和计算的。tensor我们主要关心的是它的shape和具体的value这两个属性,通过这两个属性我们可以判断我们模型是不是按照我们预先设想的方式进行运作的。通过sess.run()中fetch你指定的tensor,可以拿到我们想要观察的tensor。如下:


线性模型拟合demo

上面是一个简单的拟合一个线性模型的demo,我们通过fetch+sess.run()的方式获得了每次训练完成后的b和w,并每隔50次迭代将其type,shape和value打印出来,下面是运行结果:

step: 0 type of b:shape of b: (1,)  value of b: [ 0.67431259]type of w:shape of w: (1, 2)  value of w: [[-0.33759621  0.51075733]]

step: 50 type of b:shape of b: (1,)  value of b: [ 0.30864805]type of w:shape of w: (1, 2)  value of w: [[ 0.08913503  0.19295232]]

step: 100 type of b:shape of b: (1,)  value of b: [ 0.30055302]type of w:shape of w: (1, 2)  value of w: [[ 0.0994734  0.19937272]]

step: 150 type of b:shape of b: (1,)  value of b: [ 0.30003637]type of w:shape of w: (1, 2)  value of w: [[ 0.0999667  0.19995736]]

step: 200 type of b:shape of b: (1,)  value of b: [ 0.3000024]type of w:shape of w: (1, 2)  value of w: [[ 0.09999783  0.1999972 ]]

通过运行结果我们可以看出,每次通过sess.run返回的tensor都是一个numpy数组,这样很方便我们观察每个tensor的值,shape等,然后确定模型是不是按照我们想要的方式进行运作的。除了通过sess.run(),还可以通过tf.Printtf.Assert 的方式来观察我们想要的tensor,但是我个人还是比较喜欢用sess.run()的方式。

2.TensorFlow Debugger (tfdbg)

TensorFlow Debugger是TensorFlow自带的CLI调试工具tfdbg,这是一个很方便快捷的调试工具,官方的文档中给出了一个基于mnist数据集的一个例子,解释的很详细。下面我们还是在上面的拟合线性模型的demo上进行演示,代码如下:


tfdbg的一个demo

我们可以看到tfdbg其实很简单只需要将session包装成一个用于调试的LocalCLIDebugWrapperSession类即可,这样在session run的时候就会自动的启动tfdbg的命令行界面,进行包装的代码很简单只有一行:


将sess包装成一个用于调试的LocalCLIDebugWrapperSession类

上面代码在tfdbg运行时的命令行界面:


初始界面


sess.run()界面

通过tfdbg我们可以很方便直观的看到每一次sess.run时候的数据图中各个节点的情况,包括我们想要观察的b和w。tfdbg的功能和用法还有很多,根据需要可以自己去看官方的文档TensorFlow Debugger (tfdbg) Command-Line-Interface

3.总结

TensorFlow是一个很高级的框架,我们用它来实现深度学习算法很简洁,而且不需要考虑很底层的实现,这样有利有弊。好处是提高了开发的效率,坏处是不清楚底层的实现,有时候出现问题不能很好的找出bug所在。所以通过对模型算的debug的过程,也是进一步加深了自己对TensorFlow的理解。

个人觉得学习TensorFlow最好的方式就是自己找一个开源深度学习任务,然后自己将其复现出来,复现一定要做到指标和原有模型对齐,再这个过程中肯定会遇到各种各样的bug,但是程序员不就为了bug而生么,慢慢debug的过程,也就对TensorFlow有了更深的理解啦。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 211,948评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,371评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,490评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,521评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,627评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,842评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,997评论 3 408
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,741评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,203评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,534评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,673评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,339评论 4 330
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,955评论 3 313
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,770评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,000评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,394评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,562评论 2 349

推荐阅读更多精彩内容

  • 映礼看着我桌上的高钙奶,说,老潭你是有多缺钙。 我没理她。 她凑过来,继续说, 中国的牛奶你还信啊...
    鸣氏禾阅读 269评论 0 0
  • 1 认识一个做工程的朋友,专做新风系统,那几年政府投入4000亿做基础建设的时候,他说在北京的大街上看到塔吊就高兴...
    东成西九阅读 2,407评论 11 32
  • 重生之国民男神 异世墨莲 左手天堂,右手地狱 乔木染相思 强势攻防 龙王传说 剑凌九界 武爆仙河
    北烟邪阅读 282评论 0 0
  • 图片来自网络 1/ 读书时候,考试之前,很多人都是平时不努力,临时抱佛脚的,确实连佛脚都抱不了了,就会偶尔有人抱着...
    拼出美丽阅读 559评论 0 1