Fix container bug (iterator -> list)

Author: lucasb-eyer

DeepFried2/Container.py

@@ -22,7 +22,7 @@ def training(self):
             module.training()
 
     def parameters(self, *a, **kw):
-        params = _chain.from_iterable(m.parameters(*a, **kw) for m in self.modules)
+        params = list(_chain.from_iterable(m.parameters(*a, **kw) for m in self.modules))
 
         # We actually need to remove duplicates from the list of parameters
         # (and their corresponding gradients) in order to support reusing
@@ -31,7 +31,7 @@ def parameters(self, *a, **kw):
         return list(_OrderedDict.fromkeys(params).keys())
 
     def get_stat_updates(self):
-        return _chain.from_iterable(m.get_stat_updates() for m in self.modules)
+        return list(_chain.from_iterable(m.get_stat_updates() for m in self.modules))
 
     def add(self, *modules):
         for m in modules: