diff --git a/init.lua b/init.lua index e0fbad1..5240ba8 100644 --- a/init.lua +++ b/init.lua @@ -70,7 +70,9 @@ function M.image(img, opts) opts = opts or {} if type(img) == 'table' then - return M.images(img, opts) + local result = M.images(img, opts) + torch.setdefaulttensortype(defaultType) + return result end -- img is a collection? @@ -79,7 +81,9 @@ function M.image(img, opts) for i = 1,img:size(1) do images[i] = img[i] end - return M.images(images, opts) + local result = M.images(images, opts) + torch.setdefaulttensortype(defaultType) + return result end img = normalize(img, opts)