From b5e2022ec3bc037db1c7be279a730355955dc491 Mon Sep 17 00:00:00 2001
From: Sangram Kapre <sangram.s.kapre@gmail.com>
Date: Fri, 5 Jul 2019 10:05:26 -0700
Subject: [PATCH] Update model.py

Bug fix: Replace AttrProxy with ListModule.
---
 model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/model.py b/model.py
index d222e54..f4bcb1f 100644
--- a/model.py
+++ b/model.py
@@ -103,8 +103,8 @@ def __init__(self, state_dim, annotation_dim, n_edge_types, n_nodes,
                 m.weight.data.normal_(0.0, 0.02)
                 m.bias.data.fill_(0)
 
-        self.in_fcs = AttrProxy(self, 'in_')
-        self.out_fcs = AttrProxy(self, 'out_')
+        self.in_fcs = ListModule(self, 'in_')
+        self.out_fcs = ListModule(self, 'out_')
 
 
     def forward(self, init_hidden_state, annotation, adj_matrix):