From 708f838961992b0fd4e77f166eede06b8270c77e Mon Sep 17 00:00:00 2001 From: tgupta6 <tgupta6@illinois.edu> Date: Thu, 6 Oct 2016 17:19:55 -0500 Subject: [PATCH] removed arg space_dim from answer_classifier_cached_features/train.py and inference.py --- answer_classifier_cached_features/inference.py | 1 - answer_classifier_cached_features/train.py | 4 ---- 2 files changed, 5 deletions(-) diff --git a/answer_classifier_cached_features/inference.py b/answer_classifier_cached_features/inference.py index 7383463..1108289 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 8dc118c..af1bd70 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, -- GitLab