Hi,
Is there and elegant way to get the shape of an image in torch.utils.data.Dataset without using getitem?
Would like to use it to dynamically initialize my network module according to the image size before running on the dataset.
Thanks!