fix: revert to original way of importing

Kohulan · Jul 31, 2023 · 7af54c9 · 7af54c9
1 parent 758a0f4
commit 7af54c9
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/DECIMER/Efficient_Net_encoder.py b/DECIMER/Efficient_Net_encoder.py
@@ -1,7 +1,6 @@
 # EfficientNet-V2 config
 import tensorflow as tf
 import DECIMER.efficientnetv2 as efficientnetv2
-from DECIMER.efficientnetv2 import effnetv2_model
 
 BATCH_SIZE_DEBUG = 2
 MODEL = "efficientnetv2-m"  # @param
@@ -17,7 +16,7 @@ def get_efficientnetv2_backbone(
             "\n...At this time we only want to use the raw "
             "(no pretraining), headless, features with no pooling ...\n"
         )
-    backbone = effnetv2_model.EffNetV2Model(model_name=model_name)
+    backbone = efficientnetv2.effnetv2_model.EffNetV2Model(model_name=model_name)
     backbone(
         tf.ones((BATCH_SIZE_DEBUG, *input_shape)), training=False, features_only=True
     )