use device correctly

pyg-team · Sep 2, 2024 · d9da5eb · d9da5eb
1 parent b91cf67
commit d9da5eb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.pt_tmp/exp_version_manager.yml b/.pt_tmp/exp_version_manager.yml
@@ -1 +1 @@
-classification: 13
+classification: 15
diff --git a/benchmark/pytorch_tabular_benchmark.py b/benchmark/pytorch_tabular_benchmark.py
@@ -69,6 +69,7 @@ def train_tabular_model() -> float:
         auto_lr_find=False,
         batch_size=args.batch_size,
         max_epochs=args.epochs,
+        accelerator=str(device),
     )
 
     optimizer_config = OptimizerConfig()