fix: efficientnet imports

Kohulan · Jul 31, 2023 · 70080f6 · 70080f6
1 parent 5f607e4
commit 70080f6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/DECIMER/Efficient_Net_encoder.py b/DECIMER/Efficient_Net_encoder.py
@@ -1,8 +1,8 @@
 # EfficientNet-V2 config
 import tensorflow as tf
-import efficientnetv2
-from efficientnetv2 import effnetv2_model
-from efficientnetv2 import effnetv2_configs
+import DECIMER.efficientnetv2 as efficientnetv2
+from DECIMER.efficientnetv2 import effnetv2_model
+from DECIMER.efficientnetv2 import effnetv2_configs
 
 BATCH_SIZE_DEBUG = 2
 MODEL = "efficientnetv2-m"  # @param