Skip to content

Lite3_rl_training中Isaacgym训练及Lite3_rl_deploy部署的问题 #30

@liuzi951

Description

@liuzi951

彭博士,您好。由于电脑配置不够,我们无法根据您视频中的教程使用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训练并且部署,万分感谢!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions