Replace nosave and noval with save and val (#127)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
parent
598f17a472
commit
63c7a74691
7 changed files with 74 additions and 77 deletions
|
|
@ -67,7 +67,7 @@ def build_dataloader(cfg, batch_size, img_path, stride=32, label_path=None, rank
|
|||
augment=mode == "train", # augmentation
|
||||
hyp=cfg, # TODO: probably add a get_hyps_from_cfg function
|
||||
rect=cfg.rect if mode == "train" else True, # rectangular batches
|
||||
cache=None if cfg.noval else cfg.get("cache", None),
|
||||
cache=cfg.get("cache", None),
|
||||
single_cls=cfg.get("single_cls", False),
|
||||
stride=int(stride),
|
||||
pad=0.0 if mode == "train" else 0.5,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue