Skip to content

Commit bd99863

Browse files
committed
added coco val data for finetuning infinity_coco
1 parent 852747a commit bd99863

File tree

5 files changed

+247
-161
lines changed

5 files changed

+247
-161
lines changed

experiments/infinity_coco/hrnet/w48_384x288_adam_lr1e-3.yaml

+3-3
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ MODEL:
2626
INIT_WEIGHTS: true
2727
NAME: pose_hrnet
2828
NUM_JOINTS: 58
29-
PRETRAINED: "models/pytorch/imagenet/hrnet_w48-8ef0771d.pth"
29+
PRETRAINED: "models/pytorch/pose_coco/pose_hrnet_w48_384x288.pth"
3030
TARGET_TYPE: gaussian
3131
IMAGE_SIZE:
3232
- 288
@@ -91,7 +91,7 @@ MODEL:
9191
LOSS:
9292
USE_TARGET_WEIGHT: true
9393
TRAIN:
94-
BATCH_SIZE_PER_GPU: 10
94+
BATCH_SIZE_PER_GPU: 2
9595
SHUFFLE: true
9696
BEGIN_EPOCH: 0
9797
END_EPOCH: 200
@@ -107,7 +107,7 @@ TRAIN:
107107
MOMENTUM: 0.9
108108
NESTEROV: false
109109
TEST:
110-
BATCH_SIZE_PER_GPU: 10
110+
BATCH_SIZE_PER_GPU: 5
111111
COCO_BBOX_FILE: "data/coco/person_detection_results/COCO_val2017_detections_AP_H_56_person.json"
112112
BBOX_THRE: 1.0
113113
IMAGE_THRE: 0.0

0 commit comments

Comments
 (0)