diff --git a/constants_vision_gpu_1.py b/constants_vision_gpu_1.py index 166397aecb105daa866dcf07faacb0e911631504..578afe895932dee2d04e96f5d69d04b968269cae 100644 --- a/constants_vision_gpu_1.py +++ b/constants_vision_gpu_1.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_fixed_split' +experiment_name = 'ans_through_obj_atr_no_rel_bin_feat' ########################################################################## # Machine Specific Paths # @@ -92,6 +92,10 @@ regions_json = os.path.join( data_absolute_path, 'restructured/region_with_hypernym_labels.json') +eval_regions_json = os.path.join( + data_absolute_path, + 'restructured/region_with_labels.json') + mean_image_filename = os.path.join( data_absolute_path, 'restructured/mean_image.jpg') @@ -169,6 +173,12 @@ region_attribute_scores_dirname = os.path.join( mkdir_if_not_exists(region_attribute_scores_dirname) +region_object_scores_dirname = os.path.join( + region_output_dir, + 'object_scores') + +mkdir_if_not_exists(region_object_scores_dirname) + ########################################################################## # VQA Parameters # ########################################################################## @@ -237,14 +247,14 @@ answer_model = os.path.join( # Answer classifier training params answer_train_from_scratch = True -answer_batch_size = 25 +answer_batch_size = 50 answer_num_epochs = 20 -answer_queue_size = 500 +answer_queue_size = 200 answer_regularization_coeff = 1e-5 answer_lr = 1e-3 -answer_obj_atr_loss_wt = 1.0 -answer_ans_loss_wt = 0.0 +answer_obj_atr_loss_wt = 0.1 +answer_ans_loss_wt = 1.0 answer_mil_loss_wt = 0.0 # Answer classifier additional joint training params