Эх сурвалжийг харах

logging when using pipelines_new or models_new

gaotingquan 11 сар өмнө
parent
commit
ffac295fba

+ 3 - 1
paddlex/inference/__init__.py

@@ -12,14 +12,16 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from ..utils import flags
+from ..utils import logging
 from ..utils.flags import USE_NEW_INFERENCE, NEW_PREDICTOR
 
 if USE_NEW_INFERENCE:
+    logging.warning("=" * 20 + " Using pipelines_new " + "=" * 20)
     from .pipelines_new import create_pipeline
 else:
     from .pipelines import create_pipeline
 if NEW_PREDICTOR:
+    logging.warning("=" * 20 + " Using models_new " + "=" * 20)
     from .models_new import create_predictor
 else:
     from .models import create_predictor