configs/vg/VG_wiki_and_relco_VGG16_softmaxed_triplet_2_lan_layers_1gpu.yaml [7:50]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
  LOSS_TYPE: SOFTMAXED_TRIPLET  # EUC, SOFTMAX
  MODEL_NAME: VGG16_rel_softmaxed_triplet
  BN_MOMENTUM: 0.9
  BN_EPSILON: 1.0000001e-5
  FC_INIT_STD: 0.01
  ALLOW_INPLACE_SUM: True
  MEMONGER: True
VISUAL_EMBEDDING:
  WEIGHT_SHARING:  SBJ_OBJ_SHARED #, all_unshared,  sbj_obj_shared, all_unshared
  L2_NORMALIZE: True
TEXT_EMBEDDING:
  HIDDEN_LAYERS: 2  # 1, 0
  L2_NORMALIZE: True
INPUT_LANG_EMBEDDING_DIM: 600
OUTPUT_EMBEDDING_DIM: 1024
TRAIN:
  PARAMS_FILE: datasets/large_scale_VRD/coco_models/vgg16_model_final.pkl
  DATA_TYPE: train
  BBOX_REG: False
  BBOX_NORMALIZE_TARGETS: False  # No bbox regression at the end
  USE_FLIPPED: True
  BATCH_SIZE_PER_IM: 128
  FG_FRACTION: 0.25
  FG_THRESH: 0.5
  # SCALE_JITTER_TYPE: googleNet
  EVALUATION_FREQUENCY: 2500  # 50000
TEST:
  WEIGHTS: checkpoints/vg_wiki_and_relco/latest_final.pkl
  DATA_TYPE: test
  PROPOSAL_FILE: datasets/large_scale_VRD/proposals/vg/gt_proposals_test.pkl
  BBOX_REG: False
SOLVER:
  BASE_LR: 0.001
  STEP_SIZES: [125000]
  NUM_ITERATIONS: 175000
  WEIGHT_DECAY: 0.0005
  WEIGHT_DECAY_BN: 0.0
  MOMENTUM: 0.9
  GAMMA: 0.1
  NESTEROV: True
  WARM_UP_ITERS: 500  # 10000
  WARM_UP_LR: 0.0001
CHECKPOINT:
  CHECKPOINT_PERIOD: 25000
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



configs/vg/VG_wiki_and_relco_VGG16_softmaxed_triplet_2_lan_layers_8gpu.yaml [7:50]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
  LOSS_TYPE: SOFTMAXED_TRIPLET  # EUC, SOFTMAX
  MODEL_NAME: VGG16_rel_softmaxed_triplet
  BN_MOMENTUM: 0.9
  BN_EPSILON: 1.0000001e-5
  FC_INIT_STD: 0.01
  ALLOW_INPLACE_SUM: True
  MEMONGER: True
VISUAL_EMBEDDING:
  WEIGHT_SHARING:  SBJ_OBJ_SHARED #, all_unshared,  sbj_obj_shared, all_unshared
  L2_NORMALIZE: True
TEXT_EMBEDDING:
  HIDDEN_LAYERS: 2  # 1, 0
  L2_NORMALIZE: True
INPUT_LANG_EMBEDDING_DIM: 600
OUTPUT_EMBEDDING_DIM: 1024
TRAIN:
  PARAMS_FILE: datasets/large_scale_VRD/coco_models/vgg16_model_final.pkl
  DATA_TYPE: train
  BBOX_REG: False
  BBOX_NORMALIZE_TARGETS: False  # No bbox regression at the end
  USE_FLIPPED: True
  BATCH_SIZE_PER_IM: 128
  FG_FRACTION: 0.25
  FG_THRESH: 0.5
  # SCALE_JITTER_TYPE: googleNet
  EVALUATION_FREQUENCY: 2500  # 50000
TEST:
  WEIGHTS: checkpoints/vg_wiki_and_relco/latest_final.pkl
  DATA_TYPE: test
  PROPOSAL_FILE: datasets/large_scale_VRD/proposals/vg/gt_proposals_test.pkl
  BBOX_REG: False
SOLVER:
  BASE_LR: 0.001
  STEP_SIZES: [125000]
  NUM_ITERATIONS: 175000
  WEIGHT_DECAY: 0.0005
  WEIGHT_DECAY_BN: 0.0
  MOMENTUM: 0.9
  GAMMA: 0.1
  NESTEROV: True
  WARM_UP_ITERS: 500  # 10000
  WARM_UP_LR: 0.0001
CHECKPOINT:
  CHECKPOINT_PERIOD: 25000
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



