Hello,
Is there a way to get the size of a cuda tensor as if it were encoded as PNG?
I’ve tried this way:
inv_trans = transforms.Compose([ transforms.Normalize(mean = [ 0., 0., 0. ],
std = [ 1/0.229, 1/0.224, 1/0.225 ]),
transforms.Normalize(mean = [ -0.485, -0.456, -0.406 ],
std = [ 1., 1., 1. ]),
lambda x: x*255
])
def get_size_as_png(input_tensor):
k = inv_trans(input_tensor)
a = torchvision.io.encode_png(k.type(torch.uint8))
return sys.getsizeof(a.storage())
But when I have tensors in cuda I get the following error:
RuntimeError: Input tensor should be on CPU
Many thanks in advance