fix infinite recursion in PullTable and PushTable when converting to different types

1 files changed, 14 insertions, 0 deletions

diff --git a/PushTable.lua b/PushTable.lua
index b7cfb64..3131c69 100644
--- a/PushTable.lua
+++ b/PushTable.lua
@@ -60,4 +60,18 @@ function PushTable:updateGradInput(inputTable, gradOutputTable)
 end
 
 
+function PushTable:type(type, tensorCache)
+   assert(type, 'PullTable: must provide a type to convert to')
+
+   tensorCache = tensorCache or {}
+
+   -- find all tensors and convert them
+   for key,param in pairs(self) do
+       if(key ~= "_pulls") then
+             self[key] = nn.utils.recursiveType(param, type, tensorCache)
+        end
+   end
+   return self
+end
+