diff options
| author | Gregory Chanan <gchanan@fb.com> | 2016-12-29 00:21:36 +0300 |
|---|---|---|
| committer | Gregory Chanan <gchanan@fb.com> | 2016-12-29 00:21:36 +0300 |
| commit | 8d86d2e1dfb0d116d9ee826bb13c37d051fb4220 (patch) | |
| tree | d700dd10defa9419bc3f38aad0f39e4b330c4cc4 | |
| parent | 366ec3c30a95e4df7477ab6323564d09f7debe21 (diff) | |
Add support for HalfTensor and HalfStorage to sharedserialize.
| -rw-r--r-- | sharedserialize.lua | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/sharedserialize.lua b/sharedserialize.lua index 18612e0..fbd789a 100644 --- a/sharedserialize.lua +++ b/sharedserialize.lua @@ -86,6 +86,7 @@ for _, typename in ipairs{ 'torch.LongTensor', 'torch.FloatTensor', 'torch.DoubleTensor', + 'torch.HalfTensor', 'torch.CudaTensor', 'torch.CudaByteTensor', 'torch.CudaCharTensor', @@ -101,6 +102,7 @@ for _, typename in ipairs{ 'torch.LongStorage', 'torch.FloatStorage', 'torch.DoubleStorage', + 'torch.HalfStorage', 'torch.CudaStorage', 'torch.CudaByteStorage', 'torch.CudaCharStorage', |