Exemplos de StagedVariableGetter em Python

Exemplo n.º 1

Arquivo: variable_mgr.py Projeto: feifeibear/large-scale-tensorflow-benchmark

 def create_outer_variable_scope(self, device_num):
     self._custom_getter = variable_mgr_util.StagedVariableGetter(
         device_num, self.benchmark_cnn.raw_devices,
         self.benchmark_cnn.cpu_device, self)
     return tf.variable_scope('v',
                              reuse=bool(device_num),
                              custom_getter=self._custom_getter)

Exemplo n.º 2

Arquivo: variable_mgr.py Projeto: NanoNabla/tensorflow-benchmarks

 def create_outer_variable_scope(self, device_num):
     self._custom_getter = variable_mgr_util.StagedVariableGetter(
         device_num, self.benchmark_cnn.raw_devices, None, self)
     return tf.variable_scope('v',
                              reuse=bool(device_num) or self._reuse_vars,
                              custom_getter=self._custom_getter,
                              use_resource=self.use_resource_vars)