diff options
author | Soumith Chintala <soumith@gmail.com> | 2016-12-29 22:23:52 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2016-12-29 22:23:52 +0300 |
commit | d371549d4532a77beb935e20540d2ecbb72d0a8f (patch) | |
tree | d700dd10defa9419bc3f38aad0f39e4b330c4cc4 | |
parent | 366ec3c30a95e4df7477ab6323564d09f7debe21 (diff) | |
parent | 8d86d2e1dfb0d116d9ee826bb13c37d051fb4220 (diff) |
Merge pull request #91 from gchanan/THHalfSharedSerialize
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', |