From 66335e380dc6ac58553725beecd44c86fe981865 Mon Sep 17 00:00:00 2001 From: "Mao, Zhiyuan" Date: Thu, 20 Jan 2022 15:41:13 -0500 Subject: [PATCH] Update demo.py --- demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo.py b/demo.py index d688dde..c6beefd 100644 --- a/demo.py +++ b/demo.py @@ -53,7 +53,7 @@ # Simulate. simulator = Simulator(D/r0, im_size).to(device, dtype=torch.float32) -#Note that the current version does not support batched images. Only one frame at a time. +# Note that the current version does not support batched images. Only one frame at a time. out = simulator(x).detach().cpu().numpy() if len(out.shape) == 3: