Machine-Learning-Collection
Machine-Learning-Collection copied to clipboard
fixed 64x78 Image probllem
While i was playing with WGAN-gp i faced up RunTime Error that appeared in gradient_penalty function. It turns out transforms.Resize(64) sometimes produced images like 64x78, so change it to transforms.Resize((64, 64))