Skip to content

Commit 6263ed3

Browse files
committed
added docs for chekcpoint_path
1 parent 3fecc1f commit 6263ed3

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

synapse_net/training/supervised_training.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -244,6 +244,7 @@ def supervised_training(
244244
out_channels: The number of output channels of the UNet.
245245
mask_channel: Whether the last channels in the labels should be used for masking the loss.
246246
This can be used to implement more complex masking operations and is not compatible with `ignore_label`.
247+
checkpoint_path: Path to the directory where 'best.pt' resides; continue training this model.
247248
loader_kwargs: Additional keyword arguments for the dataloader.
248249
"""
249250
train_loader = get_supervised_loader(train_paths, raw_key, label_key, patch_shape, batch_size,

0 commit comments

Comments
 (0)