浏览代码

fix module object has no attribute NodeDef for tensorflow 0.11 (#572)

* fix module object has no attribute NodeDef for tensorflow 0.11

* change graph_pb2.NodeDef to tf.NodeDef
qingzew 9 年之前
父节点
当前提交
bf51d43420
共有 1 个文件被更改,包括 1 次插入2 次删除
  1. 1 2
      inception/inception/slim/variables.py

+ 1 - 2
inception/inception/slim/variables.py

@@ -84,7 +84,6 @@ from __future__ import print_function
 
 import tensorflow as tf
 
-from tensorflow.core.framework import graph_pb2
 from inception.slim import scopes
 
 # Collection containing all the variables created using slim.variables
@@ -211,7 +210,7 @@ def variable_device(device, name):
   """Fix the variable device to colocate its ops."""
   if callable(device):
     var_name = tf.get_variable_scope().name + '/' + name
-    var_def = graph_pb2.NodeDef(name=var_name, op='Variable')
+    var_def = tf.NodeDef(name=var_name, op='Variable')
     device = device(var_def)
   if device is None:
     device = ''