Skip to content

Commit c2ce63a

Browse files
committed
Minior fix for tar.gz
1 parent eb80725 commit c2ce63a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

efficientnetv2/effnetv2_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -777,7 +777,7 @@ def get_model(model_name,
777777

778778
if model_name in pretrained_ckpts and weights in pretrained_ckpts[model_name]:
779779
url = pretrained_ckpts[model_name][weights]
780-
fname = os.path.splitext(os.path.basename(url))[0]
780+
fname = os.path.basename(url).split('.')[0]
781781
pretrained_ckpt= tf.keras.utils.get_file(fname, url , untar=True)
782782
else:
783783
pretrained_ckpt = weights

0 commit comments

Comments
 (0)