Skip to content

Commit a6bb1ff

Browse files
fix efficientnet-l2 ckpt (#1178)
1 parent 9d7740e commit a6bb1ff

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

efficientnetv2/effnetv2_model.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -781,9 +781,12 @@ def get_model(model_name,
781781
'efficientnet-b8': {
782782
'imagenet': v1url + 'efficientnet-b8.tar.gz',
783783
},
784-
'efficientnet-l2': {
784+
'efficientnet-l2_475': {
785785
'jft': v1jfturl + 'noisy_student_efficientnet-l2_475.tar.gz',
786786
},
787+
'efficientnet-l2': {
788+
'jft': v1jfturl + 'noisy_student_efficientnet-l2.tar.gz',
789+
},
787790
}
788791

789792
if model_name in pretrained_ckpts and weights in pretrained_ckpts[model_name]:

0 commit comments

Comments
 (0)