Skip to content

Commit b6780b3

Browse files
authored
Add _v2 suffix to model file name
1 parent 90f7f59 commit b6780b3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

hubconf.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -7,11 +7,11 @@
77
# b0_transversal_5_5 is the name of entrypoint
88
def b0_transversal_5_5(**kwargs):
99
""" # This docstring shows up in hub.help()
10-
Initial model trained on 138 transversal images via transfer learning on the base model chfc-cmi/cmr-seg-tl : cmr_seg_base
10+
Initial model trained on 138 transversal images via transfer learning on the base model chfc-cmi/cmr-seg-tl : cmr_seg_base with fastai v2
1111
"""
1212
# Call the model, load pretrained weights
13-
url = "https://github.com/chfc-cmi/transversal-cmr-seg/releases/download/0.2.0/b0_transversal_5_5.pth"
14-
dst = torch.hub.get_dir() + '/chfc-cmi_transversal-cmr-seg_master/' + 'b0_transversal_5_5.pth'
13+
url = "https://github.com/chfc-cmi/transversal-cmr-seg/releases/download/0.2.0/b0_transversal_5_5_v2.pth"
14+
dst = torch.hub.get_dir() + '/chfc-cmi_transversal-cmr-seg_master/' + 'b0_transversal_5_5_v2.pth'
1515
if not os.path.isfile(dst):
1616
torch.hub.download_url_to_file(url,dst,progress=True)
1717
model = fastai.learner.load_learner(path=tempfile.gettempdir(),file=dst,tfm_y=False)

0 commit comments

Comments
 (0)