diff options
Diffstat (limited to 'ModuleFromCriterion.lua')
| -rw-r--r-- | ModuleFromCriterion.lua | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ModuleFromCriterion.lua b/ModuleFromCriterion.lua index 8775c3d..faca485 100644 --- a/ModuleFromCriterion.lua +++ b/ModuleFromCriterion.lua @@ -10,6 +10,8 @@ function ModuleFromCriterion:__init(criterion) self.gradInput = {torch.Tensor(), torch.Tensor()} end +local unpack = unpack or table.unpack -- lua52 compat + --[[ The input is a {prediction, target} pair. The output is a tensor with one number: the criterion output. --]] |