ResNetVAE
ResNetVAE copied to clipboard
adds option to freeze resnet backbone in encoder
The current ResNet_VAE constructor just loads the pretrained weights but doesn't freeze them. This PR adds an option to freeze the resnet backbone