Skip to content

Commit fb6c499

Browse files
Update README.md
1 parent 105fa25 commit fb6c499

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -239,7 +239,8 @@ check_point_stride = 30 # the checkpoint is saved for every 30 epochs.
239239
logger = Logger(model_name="pspnet_mobilenet_v2", data_name='example')
240240

241241
trainer = Trainer(model, optimizer, logger, num_epochs,
242-
train_loader, test_loader, epoch=254, check_point_epoch_stride=check_point_stride)
242+
train_loader, test_loader,
243+
check_point_epoch_stride=check_point_stride)
243244

244245
```
245246

0 commit comments

Comments
 (0)