fix val each epoch

This commit is contained in:
wa22 2024-05-25 02:05:42 +00:00
parent 74f62fb017
commit f19f3e521f

View File

@ -14,7 +14,7 @@ batch: 16 # (int) number of images per batch (-1 for AutoBatch)
imgsz: 640 # (int | list) input images size as int for train and val modes, or list[w,h] for predict and export modes imgsz: 640 # (int | list) input images size as int for train and val modes, or list[w,h] for predict and export modes
save: True # (bool) save train checkpoints and predict results save: True # (bool) save train checkpoints and predict results
save_period: -1 # (int) Save checkpoint every x epochs (disabled if < 1) save_period: -1 # (int) Save checkpoint every x epochs (disabled if < 1)
val_period: 10 # (int) Validation every x epochs val_period: 1 # (int) Validation every x epochs
cache: False # (bool) True/ram, disk or False. Use cache for data loading cache: False # (bool) True/ram, disk or False. Use cache for data loading
device: # (int | str | list, optional) device to run on, i.e. cuda device=0 or device=0,1,2,3 or device=cpu device: # (int | str | list, optional) device to run on, i.e. cuda device=0 or device=0,1,2,3 or device=cpu
workers: 8 # (int) number of worker threads for data loading (per RANK if DDP) workers: 8 # (int) number of worker threads for data loading (per RANK if DDP)