From f2df2d6a6c84f93c9eda66a67da5b904cd685302 Mon Sep 17 00:00:00 2001 From: Shiyong Ma Date: Tue, 27 Aug 2019 14:48:47 -0400 Subject: [PATCH] Update maml.py --- maml.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/maml.py b/maml.py index 92eccbd9b..4af0f1dc2 100644 --- a/maml.py +++ b/maml.py @@ -200,7 +200,7 @@ def construct_conv_weights(self): weights['b4'] = tf.Variable(tf.zeros([self.dim_hidden])) if FLAGS.datasource == 'miniimagenet': # assumes max pooling - weights['w5'] = tf.get_variable('w5', [self.dim_hidden*5*5, self.dim_output], initializer=fc_initializer) + weights['w5'] = tf.get_variable('w5', [self.dim_hidden*6*6, self.dim_output], initializer=fc_initializer) weights['b5'] = tf.Variable(tf.zeros([self.dim_output]), name='b5') else: weights['w5'] = tf.Variable(tf.random_normal([self.dim_hidden, self.dim_output]), name='w5')