diff options
author | Clement Farabet <clement.farabet@gmail.com> | 2011-07-29 01:26:37 +0400 |
---|---|---|
committer | Clement Farabet <clement.farabet@gmail.com> | 2011-07-29 01:26:37 +0400 |
commit | d7ca10c90eccbd1ab47fee193fe5db1603daf16f (patch) | |
tree | 0881edfd2d9f40e543ca18548e7a07de089092dd /OmpModule.lua | |
parent | ce4effdc67d16422ef45b2deeaab485c98b0c4ce (diff) |
Fixed multiple things: openmp and bad conv problem.
WARNING: now relies on new conv in torch.
Diffstat (limited to 'OmpModule.lua')
-rw-r--r-- | OmpModule.lua | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/OmpModule.lua b/OmpModule.lua index 73eb3c6..d7cf088 100644 --- a/OmpModule.lua +++ b/OmpModule.lua @@ -3,17 +3,17 @@ local OmpModule,parent = torch.class('nn.OmpModule','nn.Module') function OmpModule:__init() parent.__init(self) if openmp then - self.threads = openmp.getNumThreads() + self.nThread = openmp.getNumThreads() else - self.threads = 1 + self.nThread = 1 end end function OmpModule:read(file) parent.read(self, file) if openmp then - self.threads = openmp.getNumThreads() + self.nThread = openmp.getNumThreads() else - self.threads = 1 + self.nThread = 1 end end |