Augmenting (only Trainset, not Testset) Before Training: RuntimeError: inconsistent tensor sizes

4

i mean image resolution.

You can check what is the incorrect size by printing the sizes of tensors in a custom collate_fn that you supply.

3 Likes