Skip to content

Commit 2ebb67e

Browse files
committed
Update test_legacy_checkpoints.py
1 parent e04c956 commit 2ebb67e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/checkpointing/test_legacy_checkpoints.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,9 +83,9 @@ def test_resume_legacy_checkpoints(tmpdir, pl_version: str):
8383
callbacks=[es, stop],
8484
max_epochs=21,
8585
accumulate_grad_batches=2,
86-
deterministic=True,
8786
resume_from_checkpoint=path_ckpt,
8887
)
88+
torch.backends.cudnn.deterministic = True
8989
trainer.fit(model, datamodule=dm)
9090
res = trainer.test(model, datamodule=dm)
9191
assert res[0]["test_loss"] <= 0.7

0 commit comments

Comments
 (0)