fix typo

segasai · Jan 19, 2025 · 2a5e5aa · 2a5e5aa
1 parent 24d8c22
commit 2a5e5aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py/rvspecfit/nn/train_interpolator.py b/py/rvspecfit/nn/train_interpolator.py
@@ -202,7 +202,7 @@ def main(args):
 
     Tvecs0 = torch.FloatTensor(data=vecs)
     Tdat0 = torch.as_tensor(dats)
-    batch_on_dev = args.batch_on_dev
+    batch_on_dev = args.batch_on_device
 
     if not batch_on_dev:
         Tvecs0 = Tvecs0.to(train_dev)