fix: handle small input images
If input images didn't need resizing because they were already smaller than max width/height then they didn't get normalized to a multiple of 64. This caused an exception like the following: ```Sizes of tensors must match except in dimension 1. Expected size 4 but got size 3 for tensor number 1 in the list. ```pull/49/head
parent
d3623882db
commit
2221d3d1c2
Loading…
Reference in New Issue