diff --git a/constants_vision_gpu_2.py b/constants_vision_gpu_2.py index faef8795587f3ede2ec922c6ab8252765dc78c1b..04654870b9d5921c10e8d4fc59e3726c3a3bf9c8 100644 --- a/constants_vision_gpu_2.py +++ b/constants_vision_gpu_2.py @@ -5,7 +5,7 @@ def mkdir_if_not_exists(dir_name): if not os.path.exists(dir_name): os.mkdir(dir_name) -experiment_name = 'obj_atr_through_none' +experiment_name = 'obj_atr_through_mil_fixed_split' ########################################################################## # Machine Specific Paths # @@ -90,7 +90,7 @@ attribute_labels_json = os.path.join( regions_json = os.path.join( data_absolute_path, - 'restructured/region_with_labels.json') + 'restructured/region_with_hypernym_labels.json') mean_image_filename = os.path.join( data_absolute_path, @@ -110,7 +110,7 @@ genome_val_region_ids = os.path.join( genome_test_region_ids = os.path.join( data_absolute_path, - 'restructured/test_region_ids.json') + 'restructured/test_region_ids_simple.json') ########################################################################## # Genome Training/Fine-Tuning Parameters # @@ -245,7 +245,7 @@ answer_lr = 1e-3 answer_obj_atr_loss_wt = 1.0 answer_ans_loss_wt = 0.0 -answer_mil_loss_wt = 0.0 +answer_mil_loss_wt = 0.1 # Answer classifier additional joint training params num_regions_with_labels = 100