diff --git a/configs/kr_lp_pgnet.yml b/configs/kr_lp_pgnet.yml index baa306e..6d23c49 100644 --- a/configs/kr_lp_pgnet.yml +++ b/configs/kr_lp_pgnet.yml @@ -83,7 +83,7 @@ Train: channel_first: False - E2ELabelEncodeTrain: - PGProcessTrain: - batch_size: 32 + batch_size: 16 use_resize: True use_random_crop: False min_crop_size: 24 @@ -95,7 +95,7 @@ Train: loader: shuffle: True drop_last: True - batch_size_per_card: 32 + batch_size_per_card: 16 num_workers: 8 Eval: