Merge 2400a7c into 1bec122

Lightning-Universe · Mar 4, 2021 · 0c5981e · 0c5981e
2 parents 1bec122 + 2400a7c
commit 0c5981e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pl_bolts/models/autoencoders/basic_ae/basic_ae_module.py b/pl_bolts/models/autoencoders/basic_ae/basic_ae_module.py
@@ -27,7 +27,7 @@ class AE(pl.LightningModule):
         ae = AE()
 
         # pretrained on cifar10
-        ae = AE.from_pretrained('cifar10-resnet18')
+        ae = AE(input_height=32).from_pretrained('cifar10-resnet18')
     """
 
     pretrained_urls = {

diff --git a/pl_bolts/models/autoencoders/basic_vae/basic_vae_module.py b/pl_bolts/models/autoencoders/basic_vae/basic_vae_module.py
@@ -27,10 +27,10 @@ class VAE(pl.LightningModule):
         vae = VAE()
 
         # pretrained on cifar10
-        vae = VAE.from_pretrained('cifar10-resnet18')
+        vae = VAE(input_height=32).from_pretrained('cifar10-resnet18')
 
         # pretrained on stl10
-        vae = VAE.from_pretrained('stl10-resnet18')
+        vae = VAE(input_height=32).from_pretrained('stl10-resnet18')
     """
 
     pretrained_urls = {