Skip to content

GitLab

Commit 30a5aa0f authored by Alykhan Tejani's avatar Alykhan Tejani Committed by Soumith Chintala
Browse files

make pretrained densenet CPU compatible (#251)

parent c94ea5ff
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
torchvision/models/densenet.py
View file @ 30a5aa0f
...@@ -8,10 +8,10 @@ __all__ = ['DenseNet', 'densenet121', 'densenet169', 'densenet201', 'densenet161 ...@@ -8,10 +8,10 @@ __all__ = ['DenseNet', 'densenet121', 'densenet169', 'densenet201', 'densenet161
model_urls = { model_urls = {
'densenet121': 'https://download.pytorch.org/models/densenet121-241335ed.pth', 'densenet121': 'https://download.pytorch.org/models/densenet121-a639ec97.pth',
'densenet169': 'https://download.pytorch.org/models/densenet169-6f0f7f60.pth', 'densenet169': 'https://download.pytorch.org/models/densenet169-b2777c0a.pth',
'densenet201': 'https://download.pytorch.org/models/densenet201-4c113574.pth', 'densenet201': 'https://download.pytorch.org/models/densenet201-c1103571.pth',
'densenet161': 'https://download.pytorch.org/models/densenet161-17b70270.pth', 'densenet161': 'https://download.pytorch.org/models/densenet161-8d451a50.pth',
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment