diff options
Diffstat (limited to 'OnlineTrainer.lua')
-rw-r--r-- | OnlineTrainer.lua | 16 |
1 files changed, 1 insertions, 15 deletions
diff --git a/OnlineTrainer.lua b/OnlineTrainer.lua index dc6e860..920a523 100644 --- a/OnlineTrainer.lua +++ b/OnlineTrainer.lua @@ -49,9 +49,7 @@ function OnlineTrainer:log() end end print('<trainer> saving network to '..filename) - local file = torch.DiskFile(filename,'w') - self.module:write(file) - file:close() + torch.save(filename, self.module) end function OnlineTrainer:train(dataset) @@ -173,15 +171,3 @@ function OnlineTrainer:test(dataset) return self.currentError end - -function OnlineTrainer:write(file) - parent.write(self,file) - file:writeObject(self.module) - file:writeObject(self.criterion) -end - -function OnlineTrainer:read(file) - parent.read(self,file) - self.module = file:readObject() - self.criterion = file:readObject() -end |