commit a305ce3ece746575c89a69704e73dbf2eb4251ff
author Junjie Bai <jbai@fb.com> Thu Jul 13 23:28:15 2017 -0700
committer Facebook Github Bot <facebook-github-bot@users.noreply.github.com> Thu Jul 13 23:31:54 2017 -0700
tree 6975f45834fec1428b8917b8b0bb498aaa85c885
parent f44991b3980e1cdaa3c35a12a2804099e97e7942

Fix broken seq2seq example

Reviewed By: harouwu

Differential Revision: D5423060

fbshipit-source-id: 4537b020546503a1f9cb237257ab3c42665ae07f

caffe2/python/models/seq2seq/seq2seq_util.py

diff --git a/caffe2/python/models/seq2seq/seq2seq_util.py b/caffe2/python/models/seq2seq/seq2seq_util.py
index c28c137..f24a01e 100644
--- a/caffe2/python/models/seq2seq/seq2seq_util.py
+++ b/caffe2/python/models/seq2seq/seq2seq_util.py
@@ -11,7 +11,7 @@
 from future.utils import viewitems
 
 import caffe2.proto.caffe2_pb2 as caffe2_pb2
-from caffe2.python import core, rnn_cell
+from caffe2.python import core, rnn_cell, brew
 
 
 PAD_ID = 0
@@ -292,14 +292,16 @@
             initial_attention_weighted_encoder_context,
         )
     else:
-        decoder_initial_hidden_state = model.FC(
+        decoder_initial_hidden_state = brew.fc(
+            model,
             final_encoder_hidden_state,
             'decoder_initial_hidden_state',
             encoder_num_units,
             decoder_num_units,
             axis=2,
         )
-        decoder_initial_cell_state = model.FC(
+        decoder_initial_cell_state = brew.fc(
+            model,
             final_encoder_cell_state,
             'decoder_initial_cell_state',
             encoder_num_units,
@@ -320,7 +322,8 @@
     decoder_softmax_size,
 ):
     if decoder_softmax_size is not None:
-        decoder_outputs = model.FC(
+        decoder_outputs = brew.fc(
+            model,
             decoder_outputs,
             'decoder_outputs_scaled',
             dim_in=decoder_output_size,