예제 #1
0
 def set_up_with_runs(self, run_names):
   self.logdir = self.get_temp_dir()
   for run_name in run_names:
     self.generate_run(run_name)
   multiplexer = event_multiplexer.EventMultiplexer(size_guidance={
       # don't truncate my test data, please
       event_accumulator.TENSORS: self._STEPS,
   })
   multiplexer.AddRunsFromDirectory(self.logdir)
   multiplexer.Reload()
   context = base_plugin.TBContext(logdir=self.logdir, multiplexer=multiplexer)
   self.plugin = distributions_plugin.DistributionsPlugin(context)
예제 #2
0
 def set_up_with_runs(self, run_names):
     logdir = self.get_temp_dir()
     for run_name in run_names:
         self.generate_run(logdir, run_name)
     multiplexer = event_multiplexer.EventMultiplexer(
         size_guidance={
             # don't truncate my test data, please
             tag_types.TENSORS:
             self._STEPS,
         })
     multiplexer.AddRunsFromDirectory(logdir)
     multiplexer.Reload()
     provider = data_provider.MultiplexerDataProvider(multiplexer, logdir)
     context = base_plugin.TBContext(logdir=logdir, data_provider=provider)
     self.plugin = distributions_plugin.DistributionsPlugin(context)