MCPcopy
hub / github.com/hojonathanho/diffusion / _run_sampling

Method _run_sampling

diffusion_tf/tpu_utils/tpu_utils.py:333–341  ·  view source on GitHub ↗
(self, sess, ema: bool)

Source from the content-addressed store, hash-verified

331 return samples_outputs, samples_inception
332
333 def _run_sampling(self, sess, ema: bool):
334 out = {}
335 print('sampling...')
336 tstart = time.time()
337 samples = sess.run(self.ema_samples_outputs if ema else self.samples_outputs)
338 print('sampling done in {} sec'.format(time.time() - tstart))
339 for k, v in samples.items():
340 out['samples/{}'.format(k)] = v
341 return out
342
343 def _run_metrics(self, sess, ema: bool):
344 print('computing sample quality metrics...')

Callers 1

Calls 1

runMethod · 0.45

Tested by

no test coverage detected