fix: Import efficient-net properly

Kohulan · Feb 17, 2023 · 16b8e63 · 16b8e63
1 parent 6a17ee3
commit 16b8e63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/DECIMER/Efficient_Net_encoder.py b/DECIMER/Efficient_Net_encoder.py
@@ -1,6 +1,6 @@
 # EfficientNet-V2 config
 import tensorflow as tf
-import DECIMER.efficientnetv2
+import DECIMER.efficientnetv2 as efficientnetv2
 
 BATCH_SIZE_DEBUG = 2
 MODEL = "efficientnetv2-b3"  # @param