diff options
author | soumith <soumith@fb.com> | 2016-04-27 22:04:59 +0300 |
---|---|---|
committer | soumith <soumith@fb.com> | 2016-04-27 22:15:01 +0300 |
commit | 4438ec99b8ed74c1b5691b4eea12e66216e8be07 (patch) | |
tree | 85d6519f885b95269437bb08862ade0fa0966d8b /SelectTable.lua | |
parent | c8806b80ee211ce70c612addecebf236abdf8734 (diff) |
MultiLabelMarginCriterion fixes for CUDA
Diffstat (limited to 'SelectTable.lua')
-rw-r--r-- | SelectTable.lua | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/SelectTable.lua b/SelectTable.lua index d2934e5..66e71ec 100644 --- a/SelectTable.lua +++ b/SelectTable.lua @@ -55,3 +55,7 @@ function SelectTable:type(type, tensorCache) self.output = {} return parent.type(self, type, tensorCache) end + +function SelectTable:__tostring__() + return torch.type(self) .. '(' .. self.index .. ')' +end |