pocket repr's infer.py uses parameters ‘arch' with an error

[HuazhangYing]

I entered the parameters as referenced, did they change? Is there a parameter-help file?
infer.py: error: argument --arch/-a: invalid choice: 'unimol_base' (choose from )

Thanks!!!

[ZhouGengmo]

It looks like there might be a problem with the user-dir specified in the inference command, which is causing the packages not to be imported properly. You can try changing the relative path ../unimol of user-dir to an absolute path.

!CUDA_VISIBLE_DEVICES="0" python ../unimol/infer.py --user-dir ../unimol $data_path --valid-subset $job_name \
       --results-path $results_path \

[ZhouGengmo]

user-dir is for importing unimol packages. It's better not just change it to another path such as ../notebooks.