22 lines
824 B
Bash
22 lines
824 B
Bash
CONFIG_FILE=/home/qi.xiong/Temp/rdt1b-run/configs/base.yaml
|
|
PATH_TO_VISION_ENCODER_MODEL=../weights/siglip-so400m-patch14-384
|
|
PATH_TO_TEXT_ENCODER_MODEL=../weights/t5-v1_1-xxl
|
|
PATH_TO_RDT_MODEL_WIGHTS=/home/qi.xiong/Data_Qi/RDT/FTrain/checkpoints/RDT_S100_Adjust_Bottle_Clean/checkpoint-20000/pytorch_model/mp_rank_00_model_states.pt
|
|
DEVICE=cuda
|
|
CONTROL_FREQUENCY=25
|
|
PRE_LANG=False
|
|
LANG_MODEL=None
|
|
HOST=0.0.0.0
|
|
PORT=8001
|
|
|
|
python server.py \
|
|
--config_file $CONFIG_FILE \
|
|
--path_to_vision_encoder_model $PATH_TO_VISION_ENCODER_MODEL \
|
|
--path_to_text_encoder_model $PATH_TO_TEXT_ENCODER_MODEL \
|
|
--path_to_rdt_model_wights $PATH_TO_RDT_MODEL_WIGHTS \
|
|
--lang_model $LANG_MODEL \
|
|
--device $DEVICE \
|
|
--control_frequency $CONTROL_FREQUENCY \
|
|
--pre_lang $PRE_LANG \
|
|
--host $HOST \
|
|
--port $PORT |