switch auto batch size argument

app.py

@@ -209,6 +209,7 @@ def _embed(adata_bytes, layer_name, feature_col, progress):
     )
     loader = torch.utils.data.DataLoader(
         dataset,
+        batch_size=128,
         collate_fn=collate_fn,
         shuffle=False,
         drop_last=False,
@@ -221,7 +222,7 @@ def _embed(adata_bytes, layer_name, feature_col, progress):
     trainer = Trainer(
         model=model,
         device="gpu",
-        device_eval_microbatch_size="auto",
+        device_train_microbatch_size="auto",
         callbacks=[cb]
     )