TAL min_memory
argument, precommit, Docker fixes (#836)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Jaap van de Loosdrecht <jaap@vdlmv.nl>
This commit is contained in:
@ -29,6 +29,7 @@ rect: False # support rectangular training if mode='train', support rectangular
|
||||
cos_lr: False # use cosine learning rate scheduler
|
||||
close_mosaic: 10 # disable mosaic augmentation for final 10 epochs
|
||||
resume: False # resume training from last checkpoint
|
||||
min_memory: False # minimize memory footprint loss function, choices=[False, True, <roll_out_thr>]
|
||||
# Segmentation
|
||||
overlap_mask: True # masks should overlap during training (segment train only)
|
||||
mask_ratio: 4 # mask downsample ratio (segment train only)
|
||||
|
Reference in New Issue
Block a user