File tree 2 files changed +56
-0
lines changed
2 files changed +56
-0
lines changed Original file line number Diff line number Diff line change
1
+ DATASET :
2
+ NAME : " vg"
3
+ MODE : " benchmark"
4
+ TRAIN_BATCH_SIZE : 8
5
+ TEST_BATCH_SIZE : 1
6
+ MODEL :
7
+ WEIGHT_IMG : " catalog://ImageNetPretrained/MSRA/R-101"
8
+ WEIGHT_DET : " "
9
+ RELATION_ON : True
10
+ ALGORITHM : " sg_baseline"
11
+ USE_FREQ_PRIOR : False
12
+ BACKBONE :
13
+ CONV_BODY : " R-101-C4"
14
+ FREEZE_PARAMETER : False
15
+ RPN :
16
+ FREEZE_PARAMETER : False
17
+ ROI_HEADS :
18
+ BATCH_SIZE_PER_IMAGE : 512
19
+ ROI_BOX_HEAD :
20
+ NUM_CLASSES : 151
21
+ FREEZE_PARAMETER : False
22
+ ROI_RELATION_HEAD :
23
+ NUM_CLASSES : 51
24
+ SOLVER :
25
+ BASE_LR : 5e-3
26
+ MAX_ITER : 100000
27
+ STEPS : (70000,90000)
28
+ CHECKPOINT_PERIOD : 5000
Original file line number Diff line number Diff line change
1
+ DATASET :
2
+ NAME : " vg"
3
+ MODE : " benchmark"
4
+ TRAIN_BATCH_SIZE : 8
5
+ TEST_BATCH_SIZE : 1
6
+ MODEL :
7
+ WEIGHT_IMG : " catalog://ImageNetPretrained/MSRA/R-101"
8
+ WEIGHT_DET : " checkpoints/vg_benchmark_object/R-101-C4/faster_rcnn/BatchSize_6/Base_LR_0.005/checkpoint_0099999.pth"
9
+ RELATION_ON : True
10
+ ALGORITHM : " sg_baseline"
11
+ USE_FREQ_PRIOR : False
12
+ BACKBONE :
13
+ CONV_BODY : " R-101-C4"
14
+ FREEZE_PARAMETER : True
15
+ RPN :
16
+ FREEZE_PARAMETER : True
17
+ ROI_HEADS :
18
+ BATCH_SIZE_PER_IMAGE : 512
19
+ ROI_BOX_HEAD :
20
+ NUM_CLASSES : 151
21
+ FREEZE_PARAMETER : True
22
+ ROI_RELATION_HEAD :
23
+ NUM_CLASSES : 51
24
+ SOLVER :
25
+ BASE_LR : 5e-3
26
+ MAX_ITER : 40000
27
+ STEPS : (20000,30000)
28
+ CHECKPOINT_PERIOD : 5000
You can’t perform that action at this time.
0 commit comments