Added read/write functions.

1 files changed, 19 insertions, 1 deletions

diff --git a/init.lua b/init.lua
index 82def2b..073da59 100644
--- a/init.lua
+++ b/init.lua
@@ -125,7 +125,25 @@ torch.include('nnx', 'DataSet.lua')
 torch.include('nnx', 'DataList.lua')
 torch.include('nnx', 'DataSetLabelMe.lua')
 
--- helpers:
+-- torch helpers (should not be here):
+function torch.save(filename, object, mode)
+   mode = mode or 'binary'
+   local file = torch.DiskFile(filename, 'w')
+   file[mode](file)
+   file:writeObject(object)
+   file:close()
+end
+
+function torch.load(filename, mode)
+   mode = mode or 'binary'
+   local file = torch.DiskFile(filename, 'r')
+   file[mode](file)
+   local object = file:readObject()
+   file:close()
+   return object
+end
+
+-- nn helpers:
 function nnx.empty(module)
    if module.modules then
       -- find submodules in classic containers 'modules'