Make square centre-padding a model transform
Compare changes
Files
4- #39)André Anjos authored
+ 2
− 1
@@ -15,7 +15,7 @@ import torchvision.transforms
@@ -74,6 +74,7 @@ class Alexnet(pl.LightningModule):
This MR fixes issues #23 (closed) and #39 (closed), by: