-
Notifications
You must be signed in to change notification settings - Fork 24
Description
2024-08-27 09:47:07.183593: E external/local_xla/xla/stream_executor/cuda/cuda_dnn.cc:9261] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
2024-08-27 09:47:07.183746: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:607] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2024-08-27 09:47:07.572542: E external/local_xla/xla/stream_executor/cuda/cuda_blas.cc:1515] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
2024-08-27 09:47:08.428058: I tensorflow/core/platform/cpu_feature_guard.cc:182] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
2024-08-27 09:47:15.085751: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
Unable to Load Checkpoint.
None
Traceback (most recent call last):
File "/Easter2/src/predict_line.py", line 72, in
print(infer_obj.predict(args.path)) ## change the image path with the file path you want
File "Easter2/src/predict_line.py", line 58, in predict
output = self.model.predict(img)
AttributeError: 'NoneType' object has no attribute 'predict'