diff --git a/pl_bolts/models/autoencoders/basic_ae/basic_ae_module.py b/pl_bolts/models/autoencoders/basic_ae/basic_ae_module.py index ff95063aad..66c3570dc6 100644 --- 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 index 0b2d45f09d..19e193d1e3 100644 --- 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__(