Ejemplo n.º 1
0
 def _test_step(self, inputs):
     return strategy_run(self._strategy, self.test_batch, (inputs, ))
Ejemplo n.º 2
0
 def _simple_eval_step(self, batch):
     return local_results(
         strategy_run(self._strategy, self.simple_eval_batch, (batch, )),
         self._strategy)
Ejemplo n.º 3
0
 def _train_step_on_batches(self, batches):
     strategy_run(self._strategy, self.train_batches, batches)
Ejemplo n.º 4
0
 def _eval_step(self, batch):
     strategy_run(self._strategy, self.eval_batch, (batch, ))
Ejemplo n.º 5
0
 def _train_step(self, batch):
     strategy_run(self._strategy, self.train_batch, (batch, ))