Skip to content

Commit 57e0f64

Browse files
authored
Update README.md
1 parent 264c6bb commit 57e0f64

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -401,7 +401,7 @@ from rfdetr import RFDETRBase
401401

402402
model = RFDETRBase()
403403

404-
model.train(dataset_dir=<DATASET_PATH>, epochs=10, batch_size=4, grad_accum_steps=4, lr=1e-4, output_dir=<OUTPUT_PATH>, )
404+
model.train(dataset_dir=<DATASET_PATH>, epochs=10, batch_size=4, grad_accum_steps=4, lr=1e-4, output_dir=<OUTPUT_PATH>, resume=<CHECKPOINT_PATH>)
405405
```
406406

407407
### Early stopping

0 commit comments

Comments
 (0)