diff options
| author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-02-08 03:22:07 +0300 |
|---|---|---|
| committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-02-09 21:42:51 +0300 |
| commit | 4696a46212ffc5bb54b0d8f3327dfa47693bbfdf (patch) | |
| tree | 5f4512a8ead3d952d17a9a82c200a60ba6cb5082 /BCECriterion.lua | |
| parent | 948ac6a26cc6c2812e04718911bca9a4b641020e (diff) | |
lazy init
Diffstat (limited to 'BCECriterion.lua')
| -rw-r--r-- | BCECriterion.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/BCECriterion.lua b/BCECriterion.lua index 2c614b3..b319335 100644 --- a/BCECriterion.lua +++ b/BCECriterion.lua @@ -30,7 +30,7 @@ function BCECriterion:updateOutput(input, target) assert( input:nElement() == target:nElement(), "input and target size mismatch") - self.buffer = self.buffer or torch.Tensor():typeAs(input) + self.buffer = self.buffer or input.new() local buffer = self.buffer local weights = self.weights @@ -74,7 +74,7 @@ function BCECriterion:updateGradInput(input, target) assert( input:nElement() == target:nElement(), "input and target size mismatch") - self.buffer = self.buffer or torch.Tensor():typeAs(input) + self.buffer = self.buffer or input.new() local buffer = self.buffer local weights = self.weights |