Skip to content

Commit 192e5c5

Browse files
authored
fix legacy creation (#16282)
1 parent d21e81f commit 192e5c5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/legacy/simple_classif_training.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@ def main_train(dir_path, max_epochs: int = 20):
2828
seed_everything(42)
2929
stopping = EarlyStopping(monitor="val_acc", mode="max", min_delta=0.005)
3030
trainer = pl.Trainer(
31+
accelerator="auto",
3132
default_root_dir=dir_path,
32-
devices=int(torch.cuda.is_available()),
3333
precision=(16 if torch.cuda.is_available() else 32),
3434
callbacks=[stopping],
3535
min_epochs=3,

0 commit comments

Comments
 (0)