Update AE and VAE documentation (#557)

* Update VAE documentation Previously, pretrained models were initialized with a static method. Now, they are initialized with an instance method. Upates docstring accordingly. * update documentation for AE * Apply suggestions from code review Co-authored-by: jeremyfisher <jeremyadamsfisher+git@gmail.com> Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
Lightning-Universe · Mar 6, 2021 · 261a098 · 261a098
1 parent 9ab43c4
commit 261a098
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 = {