VAE-ResNet18-PyTorch
VAE-ResNet18-PyTorch copied to clipboard
Increase layer
@julianstastny I tried to increase from 4 layers into 6 layers from your model. However I always get Cuda memory error. Do you know how to solve this? Thank you
U need a bigger GPU