Skip to content

Commit e69afd4

Browse files
committed
Fixed COCO scene images config
1 parent aa2184e commit e69afd4

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

configs/coco_scene_images_transformer.yaml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,8 @@ model:
4747
data:
4848
target: main.DataModuleFromConfig
4949
params:
50-
batch_size: 24
50+
batch_size: 6
51+
num_workers: 12
5152
train:
5253
target: taming.data.annotated_objects_coco.AnnotatedObjectsCoco
5354
params:
@@ -62,7 +63,7 @@ data:
6263
crop_method: random-1d
6364
random_flip: true
6465
use_group_parameter: true
65-
encode_crop: false
66+
encode_crop: true
6667
validation:
6768
target: taming.data.annotated_objects_coco.AnnotatedObjectsCoco
6869
params:

0 commit comments

Comments
 (0)