首页 PaddleX 帖子详情
预测提示错误不会搞 已解决
收藏
快速回复
PaddleX 问答入门 802 10
预测提示错误不会搞 已解决
收藏
快速回复
PaddleX 问答入门 802 10
# 批量预测
import numpy as np
model = pdx.load_model('output/ResNet50_vd_ssld/best_model')
result = model.batch_predict(val_list)
print("Predict Result:", result)

提示错误

2021-05-29 13:32:01 [INFO]	Model[ResNet50_vd_ssld] loaded.
---------------------------------------------------------------------------RuntimeError                              Traceback (most recent call last) in 
      2 import numpy as np
      3 model = pdx.load_model('output/ResNet50_vd_ssld/best_model')
----> 4 result = model.batch_predict(val_list)
      5 print("Predict Result:", result)
/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddlex/cv/models/classifier.py in batch_predict(self, img_file_list, transforms, topk)
    397                                   feed={'image': im},
    398                                   fetch_list=list(self.test_outputs.values()),
--> 399                                   use_program_cache=True)
    400 
    401         preds = BaseClassifier._postprocess(result, true_topk, self.labels)
/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/executor.py in run(self, program, feed, fetch_list, feed_var_name, fetch_var_name, scope, return_numpy, use_program_cache, return_merged, use_prune)
   1108                 return_merged=return_merged)
   1109         except Exception as e:
-> 1110             six.reraise(*sys.exc_info())
   1111 
   1112     def _run_impl(self, program, feed, fetch_list, feed_var_name,
/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/six.py in reraise(tp, value, tb)
    701             if value.__traceback__ is not tb:
    702                 raise value.with_traceback(tb)
--> 703             raise value
    704         finally:
    705             value = None
/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/executor.py in run(self, program, feed, fetch_list, feed_var_name, fetch_var_name, scope, return_numpy, use_program_cache, return_merged, use_prune)
   1106                 use_program_cache=use_program_cache,
   1107                 use_prune=use_prune,
-> 1108                 return_merged=return_merged)
   1109         except Exception as e:
   1110             six.reraise(*sys.exc_info())
/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/executor.py in _run_impl(self, program, feed, fetch_list, feed_var_name, fetch_var_name, scope, return_numpy, use_program_cache, return_merged, use_prune)
   1236                 scope=scope,
   1237                 return_numpy=return_numpy,
-> 1238                 use_program_cache=use_program_cache)
   1239 
   1240         program._compile(scope, self.place)
/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/executor.py in _run_program(self, program, feed, fetch_list, feed_var_name, fetch_var_name, scope, return_numpy, use_program_cache)
   1329         else:
   1330             self._default_executor.run_prepared_ctx(ctx, scope, False, False,
-> 1331                                                     False)
   1332         arr = scope.find_var(fetch_var_name).get_fetch_list()
   1333         tensors = arr._move_to_list()
RuntimeError: ResourceExhaustedError: 

Out of memory error on GPU 0. Cannot allocate 12.500000GB memory on GPU 0, available memory is only 8.557495GB.

Please check whether there is any other process using GPU 0.
1. If yes, please stop them, or start PaddlePaddle on another GPU.
2. If no, please decrease the batch size of your model. 

 (at /paddle/paddle/fluid/memory/allocation/cuda_allocator.cc:69)
JavaRoom
已解决
10# 回复于2021-06
不不不, 然后我就for循环,多次单个预测了。
0
收藏
回复
全部评论(10)
时间顺序
JavaRoom
#2 回复于2021-05

抱歉截图发不上来,被打了。

0
回复
l
laibaohua_2014
#3 回复于2021-06

显存不够

0
回复
JavaRoom
#4 回复于2021-06
显存不够

对,但是预测也花太多显存我就抑郁了。

0
回复
FutureSI
#5 回复于2021-06

小batch size吧

0
回复
JavaRoom
#6 回复于2021-06
小batch size吧

我怀疑当时我的模型有问题

0
回复
飞桨_Paddle3D
#7 回复于2021-06

预测的时候传入了多少张?

0
回复
JavaRoom
#8 回复于2021-06
预测的时候传入了多少张?

不知道得,批量预测,自动控制得?我没设置。

0
回复
FutureSI
#9 回复于2021-06

怀疑模型就试试继续train下,看看可以不

0
回复
JavaRoom
#10 回复于2021-06
怀疑模型就试试继续train下,看看可以不

不不不, 然后我就for循环,多次单个预测了。

0
回复
FutureSI
#11 回复于2021-06
JavaRoom #10
不不不, 然后我就for循环,多次单个预测了。

我试过无论继续训练还是预测,修改batch size 都是不受影响的

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户