Skip to content

Commit 459187f

Browse files
committed
Update benchmarks
1 parent 952429a commit 459187f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

benchmarks/test_basic_parity.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,8 +160,8 @@ def lightning_loop(cls_model, idx, device_type: str = "cuda", num_epochs=10):
160160
max_epochs=num_epochs if idx > 0 else 1,
161161
enable_progress_bar=False,
162162
enable_model_summary=False,
163+
enable_checkpointing=False,
163164
gpus=1 if device_type == "cuda" else 0,
164-
checkpoint_callback=False,
165165
logger=False,
166166
replace_sampler_ddp=False,
167167
)

0 commit comments

Comments
 (0)