Merge pull request #337 from torch/seqfix

fixing Sequential.remove corner case

1 files changed, 7 insertions, 2 deletions

diff --git a/Sequential.lua b/Sequential.lua
index 359a764..aead092 100644
--- a/Sequential.lua
+++ b/Sequential.lua
@@ -29,8 +29,13 @@ function Sequential:remove(index)
       error"index out of range"
    end
    table.remove(self.modules, index)
-   self.output = self.modules[#self.modules].output
-   self.gradInput = self.modules[1].gradInput
+   if #self.modules > 0 then
+       self.output = self.modules[#self.modules].output
+       self.gradInput = self.modules[1].gradInput
+   else
+       self.output = torch.Tensor()
+       self.gradInput = torch.Tensor()
+   end
 end
 
 function Sequential:updateOutput(input)