本文整理匯總了Python中inception.slim.variables.global_step方法的典型用法代碼示例。如果您正苦於以下問題:Python variables.global_step方法的具體用法?Python variables.global_step怎麽用?Python variables.global_step使用的例子?那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類inception.slim.variables
的用法示例。
在下文中一共展示了variables.global_step方法的5個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: testDeviceFn
# 需要導入模塊: from inception.slim import variables [as 別名]
# 或者: from inception.slim.variables import global_step [as 別名]
def testDeviceFn(self):
class DevFn(object):
def __init__(self):
self.counter = -1
def __call__(self, op):
self.counter += 1
return '/cpu:%d' % self.counter
with tf.Graph().as_default():
with scopes.arg_scope([variables.global_step], device=DevFn()):
gs = variables.global_step()
gs2 = variables.global_step()
self.assertDeviceEqual(gs.device, '/cpu:0')
self.assertEquals(gs, gs2)
self.assertDeviceEqual(gs2.device, '/cpu:0')
示例2: testStable
# 需要導入模塊: from inception.slim import variables [as 別名]
# 或者: from inception.slim.variables import global_step [as 別名]
def testStable(self):
with tf.Graph().as_default():
gs = variables.global_step()
gs2 = variables.global_step()
self.assertTrue(gs is gs2)
示例3: testDevice
# 需要導入模塊: from inception.slim import variables [as 別名]
# 或者: from inception.slim.variables import global_step [as 別名]
def testDevice(self):
with tf.Graph().as_default():
with scopes.arg_scope([variables.global_step], device='/gpu:0'):
gs = variables.global_step()
self.assertDeviceEqual(gs.device, '/gpu:0')
示例4: testReplicaDeviceSetter
# 需要導入模塊: from inception.slim import variables [as 別名]
# 或者: from inception.slim.variables import global_step [as 別名]
def testReplicaDeviceSetter(self):
device_fn = tf.train.replica_device_setter(2)
with tf.Graph().as_default():
with scopes.arg_scope([variables.global_step], device=device_fn):
gs = variables.global_step()
gs2 = variables.global_step()
self.assertEquals(gs, gs2)
self.assertDeviceEqual(gs.device, '/job:ps/task:0')
self.assertDeviceEqual(gs.initial_value.device, '/job:ps/task:0')
self.assertDeviceEqual(gs2.device, '/job:ps/task:0')
self.assertDeviceEqual(gs2.initial_value.device, '/job:ps/task:0')
示例5: testVariableWithVariableDeviceChooser
# 需要導入模塊: from inception.slim import variables [as 別名]
# 或者: from inception.slim.variables import global_step [as 別名]
def testVariableWithVariableDeviceChooser(self):
with tf.Graph().as_default():
device_fn = variables.VariableDeviceChooser()
with scopes.arg_scope([variables.global_step], device=device_fn):
gs = variables.global_step()
gs2 = variables.global_step()
self.assertEquals(gs, gs2)
self.assertDeviceEqual(gs.device, 'cpu:0')
self.assertDeviceEqual(gs.initial_value.device, gs.device)
self.assertDeviceEqual(gs2.device, 'cpu:0')
self.assertDeviceEqual(gs2.initial_value.device, gs2.device)