back-compatibility of VolumetricMaxPooling

1 files changed, 9 insertions, 0 deletions

diff --git a/VolumetricMaxPooling.lua b/VolumetricMaxPooling.lua
index 5e5cb04..1d6ca6d 100644
--- a/VolumetricMaxPooling.lua
+++ b/VolumetricMaxPooling.lua
@@ -1,5 +1,7 @@
 local VolumetricMaxPooling, parent = torch.class('nn.VolumetricMaxPooling', 'nn.Module')
 
+VolumetricMaxPooling.__version = 2
+
 function VolumetricMaxPooling:__init(kT, kW, kH, dT, dW, dH)
    parent.__init(self)
 
@@ -46,3 +48,10 @@ function VolumetricMaxPooling:empty()
    self.indices:resize()
    self.indices:storage():resize(0)
 end
+
+function VolumetricMaxPooling:read(file, version)
+   parent.read(self, file)
+   if version < 2 then
+      self.ceil_mode = false
+   end
+end