display
display copied to clipboard
Fix restoring default tensor type after call .image(...)
Non-working example:
torch.setdefaulttensortype('torch.FloatTensor')
print(torch.getdefaulttensortype())
display = require('display')
display.image({torch.Tensor(1, 1, 1)})
print(torch.getdefaulttensortype())
Output:
torch.FloatTensor
torch.DoubleTensor
After fix and correct output:
torch.FloatTensor
torch.FloatTensor
@szym do merge as soon as can plz.