Merge 1332697 into 1bec122

Lightning-Universe · Feb 28, 2021 · b046c1e · b046c1e
2 parents 1bec122 + 1332697
commit b046c1e
Showing 1 changed file with 1 addition and 1 deletion.
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 = {