预测提示错误不会搞
收藏
# 批量预测
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)
0
收藏
请登录后评论
抱歉截图发不上来,被打了。
显存不够
对,但是预测也花太多显存我就抑郁了。
小batch size吧
我怀疑当时我的模型有问题
预测的时候传入了多少张?
不知道得,批量预测,自动控制得?我没设置。
怀疑模型就试试继续train下,看看可以不
不不不, 然后我就for循环,多次单个预测了。
我试过无论继续训练还是预测,修改batch size 都是不受影响的