def test_video(self): try: import moviepy # noqa F401 except ImportError: return self.assertTrue(compare_proto(summary.video('dummy', np.random.rand(4, 3, 1, 8, 8).astype(np.float32)), self)) summary.video('dummy', np.random.rand(16, 48, 1, 28, 28).astype(np.float32)) summary.video('dummy', np.random.rand(20, 7, 1, 8, 8).astype(np.float32))
def test_video(self): try: import moviepy # noqa: F401 except ImportError: return self.assertTrue(compare_proto(summary.video('dummy', tensor_N(shape=(4, 3, 1, 8, 8))), self)) summary.video('dummy', np.random.rand(16, 48, 1, 28, 28)) summary.video('dummy', np.random.rand(20, 7, 1, 8, 8))
def add_video(self, tag, vid_tensor, global_step=None, fps=4, walltime=None): torch._C._log_api_usage_once("tensorboard.logging.add_video") self._get_file_writer().add_summary(video(tag, vid_tensor, fps), global_step, walltime)