diff --git a/answer_classifier_cached_features/inference.py b/answer_classifier_cached_features/inference.py index 73834634eb9a5ecf2af136da8d263e3ac04fc84e..11082891125a6565fa6628dfa7a141ecc538a203 100644 --- a/answer_classifier_cached_features/inference.py +++ b/answer_classifier_cached_features/inference.py @@ -14,7 +14,6 @@ class AnswerInference(): noun_embed, adjective_embed, num_answers, - space_dim, keep_prob, is_training=True): diff --git a/answer_classifier_cached_features/train.py b/answer_classifier_cached_features/train.py index 8dc118c14f6a3c6b16ca4f8f0695ecc9c081223a..af1bd708e191eae0108ee0ca86dd9063f5cc8002 100644 --- a/answer_classifier_cached_features/train.py +++ b/answer_classifier_cached_features/train.py @@ -24,7 +24,6 @@ class graph_creator(): batch_size, image_size, num_neg_answers, - space_dim, regularization_coeff, num_regions_wo_labels, num_regions_w_labels, @@ -37,7 +36,6 @@ class graph_creator(): training=True): self.im_h, self.im_w = image_size self.num_neg_answers = num_neg_answers - self.space_dim = space_dim self.batch_size = batch_size self.regularization_coeff = regularization_coeff self.num_regions_wo_labels = num_regions_wo_labels @@ -112,7 +110,6 @@ class graph_creator(): self.noun_embed, self.adjective_embed, self.num_neg_answers + 1, - self.space_dim, self.plh['keep_prob'], self.training) @@ -913,7 +910,6 @@ if __name__=='__main__': constants.answer_batch_size, constants.image_size, constants.num_negative_answers, - constants.answer_embedding_dim, constants.answer_regularization_coeff, constants.answer_batch_size*constants.num_region_proposals, constants.num_regions_with_labels,