Add noisy-student weights for b0-b7 models

qubvel · Feb 28, 2020 · 01cc666 · 01cc666
1 parent aa1edca
commit 01cc666
Show file tree

Hide file tree

Showing 4 changed files with 221 additions and 140 deletions.
diff --git a/efficientnet/keras.py b/efficientnet/keras.py
@@ -11,6 +11,7 @@
 EfficientNetB5 = inject_keras_modules(model.EfficientNetB5)
 EfficientNetB6 = inject_keras_modules(model.EfficientNetB6)
 EfficientNetB7 = inject_keras_modules(model.EfficientNetB7)
+EfficientNetL2 = inject_keras_modules(model.EfficientNetL2)
 
 preprocess_input = inject_keras_modules(model.preprocess_input)