模型超过2G怎么处理 #32
Description
File "/workdir/common/anaconda3/envs/qpixart/lib/python3.9/runpy.py", line 87, in _run_code
exec(code, run_globals)
File "/mnt/dolphinfs/hdd_pool/docker/user/hadoop-platcv/tsl/q_mtpixart/Dipoorlet/dipoorlet/main.py", line 120, in
act_clip_val, weight_clip_val = tensor_calibration(onnx_graph, args)
File "/mnt/dolphinfs/hdd_pool/docker/user/hadoop-platcv/tsl/q_mtpixart/Dipoorlet/dipoorlet/tensor_cali/tensor_cali_base.py", line 6, in tensor_calibration
act_clip_val = tensor_cali_dispatcher(args.act_quant, onnx_graph, args)
File "/mnt/dolphinfs/hdd_pool/docker/user/hadoop-platcv/tsl/q_mtpixart/Dipoorlet/dipoorlet/utils.py", line 297, in wrapper
return dispatch(args[0])(*(args[1:]), **kw)
File "/mnt/dolphinfs/hdd_pool/docker/user/hadoop-platcv/tsl/q_mtpixart/Dipoorlet/dipoorlet/tensor_cali/basic_algorithm.py", line 35, in find_clip_val_hist
stats_min_max = forward_get_minmax(onnx_graph, args)
File "/mnt/dolphinfs/hdd_pool/docker/user/hadoop-platcv/tsl/q_mtpixart/Dipoorlet/dipoorlet/forward_net.py", line 200, in forward_get_minmax
ort_session = ort.InferenceSession(net.SerializeToString(), providers=providers)