Skip to content
Snippets Groups Projects
Commit d97f23e7 authored by tgupta6's avatar tgupta6
Browse files

updated constants_vision_gpu_2

parent d3b9aaf6
No related branches found
No related tags found
No related merge requests found
......@@ -5,7 +5,7 @@ def mkdir_if_not_exists(dir_name):
if not os.path.exists(dir_name):
os.mkdir(dir_name)
experiment_name = 'QA_explicit_dot_pretrained_same_lr'
experiment_name = 'QA_explicit_dot_joint_training_pretrained_multi_rate'
#experiment_name = 'object_attribute_classifier_large_images'
# Global output directory (all subexperiments will be saved here)
global_output_dir = '/data/tanmay/GenVQA_Exp_Results'
......@@ -152,21 +152,22 @@ vqa_answer_vocab_json = os.path.join(
'answer_vocab.json')
# VQA dataset params
num_train_questions = 248349
num_val_subset_questions = 10000
num_val_questions = 121512
num_val_rest_questions = num_val_questions - num_val_subset_questions
num_test_questions = 0
# num_train_questions = 248349
# num_val_subset_questions = 10000
# num_val_questions = 121512
# num_val_rest_questions = num_val_questions - num_val_subset_questions
# num_test_questions = 0
# Answer classifier training params
answer_batch_size = 50
answer_num_epochs = 10
answer_offset = 0
answer_obj_atr_loss_wt = 0.0
answer_obj_atr_loss_wt = 0.1
answer_regularization_coeff = 1e-5
answer_queue_size = 500
answer_embedding_dim = 600
answer_lr = 1e-4
obj_atr_lr = 1e-4
answer_lr = 1e-3
answer_log_every_n_iter = 500
answer_output_dir = os.path.join(
global_experiment_dir,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment