Merge branch 'master' into patch-1

Lightning-Universe · Jun 15, 2021 · 3156315 · 3156315
2 parents 21790d3 + 73a3223
commit 3156315
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 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
@@ -1,4 +1,4 @@
-import os
+import urllib.parse
 from argparse import ArgumentParser
 
 import pytorch_lightning as pl
@@ -31,7 +31,7 @@ class AE(pl.LightningModule):
     """
 
     pretrained_urls = {
-        'cifar10-resnet18': os.path.join(_HTTPS_AWS_HUB, 'ae/ae-cifar10/checkpoints/epoch%3D96.ckpt'),
+        'cifar10-resnet18': urllib.parse.urljoin(_HTTPS_AWS_HUB, 'ae/ae-cifar10/checkpoints/epoch%3D96.ckpt'),
     }
 
     def __init__(

diff --git a/pl_bolts/models/autoencoders/basic_vae/basic_vae_module.py b/pl_bolts/models/autoencoders/basic_vae/basic_vae_module.py
@@ -1,4 +1,4 @@
-import os
+import urllib.parse
 from argparse import ArgumentParser
 
 import pytorch_lightning as pl
@@ -34,8 +34,8 @@ class VAE(pl.LightningModule):
     """
 
     pretrained_urls = {
-        'cifar10-resnet18': os.path.join(_HTTPS_AWS_HUB, 'vae/vae-cifar10/checkpoints/epoch%3D89.ckpt'),
-        'stl10-resnet18': os.path.join(_HTTPS_AWS_HUB, 'vae/vae-stl10/checkpoints/epoch%3D89.ckpt'),
+        'cifar10-resnet18': urllib.parse.urljoin(_HTTPS_AWS_HUB, 'vae/vae-cifar10/checkpoints/epoch%3D89.ckpt'),
+        'stl10-resnet18': urllib.parse.urljoin(_HTTPS_AWS_HUB, 'vae/vae-stl10/checkpoints/epoch%3D89.ckpt'),
     }
 
     def __init__(