FileNotFoundError: [Errno 2] No such file or directory: '/workspace/checkpoints/checkpoint_2/inference.config.json'

[2561681244]

Hello!

This is my running code:

CUDA_VISIBLE_DEVICES=3 python3 scripts/robotwin_eval.py
--model_dir /mnt/nvme1/yfyuan/wangxiao/RoboOrchardLab/projects/holobrain/workspace/checkpoints/checkpoint_2
--task_names ["move_stapler_pad"]
--mode ray
--device cuda
--gpu_ids [3]
--workers_per_gpu 1

However,I encountered this problem:

...

And here is my workspace:

[AlanHuangT]

@jin-yiwei

[jin-yiwei]

Hi @2561681244
The directory corresponding to model_dir should contain the model configuration, model weights, and inference configuration (for the scripts we provide, the default prefix for the inference configuration is inference.config). If you are using one of our provided models in huggingface like here, the robotwin2_0_config.json or robotwin2_0_ur5_wsg.config.json indicates the specific prefix for the various configurations, which you will need to adjust accordingly; if you are using your own model, you must use the export script to generate and export the inference configuration.

Additionally, I noticed that we had omitted the CLI entry point for the inference prefix. We have add this cli parameter, you can now use the latest code and specify the inference config prefix.