Skip to content

Commit 24836b8

Browse files
committed
changed experiments config
1 parent bd99863 commit 24836b8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

experiments/infinity_coco/hrnet/w48_384x288_adam_lr1e-3.yaml

+2-2
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ DATASET:
1717
FLIP: true
1818
NUM_JOINTS_HALF_BODY: 27
1919
PROB_HALF_BODY: 0.3
20-
ROOT: "../infinity_dataset_medium_512"
20+
ROOT: "../infinity_dataset_combined"
2121
ROT_FACTOR: 45
2222
SCALE_FACTOR: 0.35
2323
TEST_SET: "test"
@@ -91,7 +91,7 @@ MODEL:
9191
LOSS:
9292
USE_TARGET_WEIGHT: true
9393
TRAIN:
94-
BATCH_SIZE_PER_GPU: 2
94+
BATCH_SIZE_PER_GPU: 20
9595
SHUFFLE: true
9696
BEGIN_EPOCH: 0
9797
END_EPOCH: 200

0 commit comments

Comments
 (0)