彭博士,您好。由于电脑配置不够,我们无法根据您视频中的教程使用Isaacsim训练,只能采用过去的isaacgym进行训练[Lite3_rl_training],参照例程Lite3_rl_deploy。请问是否能提供一个可用的pt2onnx.py文件呢。现在的pt2onnx.py跟前一位用户[toolren]出现相同的问题,无法用pt2onnx.py直接对[Lite3_rl_deploy]训练出的pt文件直接转换。
问题如下:
python pt2onnx.py
Loading TorchScript model from model_15000.pt
Traceback (most recent call last):
File "pt2onnx.py", line 11, in
ts_model = torch.jit.load(pt_model_path)
File "/home/lzq/miniconda3/envs/dr_gym/lib/python3.8/site-packages/torch/jit/_serialization.py", line 158, in load
cpp_module = torch._C.import_ir_module(cu, os.fspath(f), map_location, _extra_files, _restore_shapes) # type: ignore[call-arg]
RuntimeError: PytorchStreamReader failed locating file constants.pkl: file not found
麻烦您有空的时候帮忙修改一下旧版的pt2onnx.py,以便我们可以通过isaacgym训练并且部署,万分感谢!
彭博士,您好。由于电脑配置不够,我们无法根据您视频中的教程使用Isaacsim训练,只能采用过去的isaacgym进行训练[Lite3_rl_training],参照例程Lite3_rl_deploy。请问是否能提供一个可用的pt2onnx.py文件呢。现在的pt2onnx.py跟前一位用户[toolren]出现相同的问题,无法用pt2onnx.py直接对[Lite3_rl_deploy]训练出的pt文件直接转换。
问题如下:
python pt2onnx.py
Loading TorchScript model from model_15000.pt
Traceback (most recent call last):
File "pt2onnx.py", line 11, in
ts_model = torch.jit.load(pt_model_path)
File "/home/lzq/miniconda3/envs/dr_gym/lib/python3.8/site-packages/torch/jit/_serialization.py", line 158, in load
cpp_module = torch._C.import_ir_module(cu, os.fspath(f), map_location, _extra_files, _restore_shapes) # type: ignore[call-arg]
RuntimeError: PytorchStreamReader failed locating file constants.pkl: file not found
麻烦您有空的时候帮忙修改一下旧版的pt2onnx.py,以便我们可以通过isaacgym训练并且部署,万分感谢!