From f9ab19ffe5ef2f8f3780d01b0eea836840358a0d Mon Sep 17 00:00:00 2001
From: hsmaan <hassaan.maan@mail.utoronto.ca>
Date: Mon, 22 Jan 2024 16:11:22 -0500
Subject: [PATCH] initial commit

---
 deepvelo/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deepvelo/train.py b/deepvelo/train.py
index 980530d..1668378 100644
--- a/deepvelo/train.py
+++ b/deepvelo/train.py
@@ -86,7 +86,7 @@ def train(
     callback: Callable = None,
     **kwargs,
 ):
-    batch_size, n_genes = adata.layers["Ms"].shape
+    batch_size, n_genes = adata.layers["Ms"].shape # To be changed after mini-batch training is implemented
     if configs["data_loader"]["args"]["velocity_genes"]:
         n_genes = int(np.sum(adata.var["velocity_genes"]))
     configs["arch"]["args"]["n_genes"] = n_genes