Update demo.py

StanleyChanGroup · Jan 20, 2022 · 66335e3 · 66335e3
1 parent fdb509e
commit 66335e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: