We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent eb80725 commit c2ce63aCopy full SHA for c2ce63a
efficientnetv2/effnetv2_model.py
@@ -777,7 +777,7 @@ def get_model(model_name,
777
778
if model_name in pretrained_ckpts and weights in pretrained_ckpts[model_name]:
779
url = pretrained_ckpts[model_name][weights]
780
- fname = os.path.splitext(os.path.basename(url))[0]
+ fname = os.path.basename(url).split('.')[0]
781
pretrained_ckpt= tf.keras.utils.get_file(fname, url , untar=True)
782
else:
783
pretrained_ckpt = weights
0 commit comments