From fdb509e627f3ab5abda8b39c20a129578bdacd5c Mon Sep 17 00:00:00 2001
From: "Mao, Zhiyuan" <mao114@purdue.edu>
Date: Thu, 20 Jan 2022 15:41:02 -0500
Subject: [PATCH] Update demo.py

---
 demo.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/demo.py b/demo.py
index c72d0a8..d688dde 100644
--- a/demo.py
+++ b/demo.py
@@ -50,9 +50,10 @@
 # Generate correlation matrix for tilt. Do this once for each different turbulence parameter. 
 tilt_mat(im_size, D, r0, L)
 
-# Simulate
+# 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. 
 out = simulator(x).detach().cpu().numpy()
 
 if len(out.shape) == 3: