Fix pretrained_urls on windows (#652)

Lightning-Universe · Jun 15, 2021 · 73a3223 · 73a3223
1 parent 49b728d
commit 73a3223
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__(