feature: train & infer scripts

Author: YoucanBaby

script/infer.sh

@@ -0,0 +1,14 @@
+#!/bin/bash
+
+CKPT_PATH="pretrained_model/wam-flow/navsim"
+FUDOKI_PATH="pretrained_model/fudoki"
+IMAGE_PATH="data/navsim_data/sensor_blobs/test/2021.09.09.17.18.51_veh-48_00889_01147/CAM_F0/9a6f0331d98258a0.jpg"
+
+torchrun --nproc_per_node 1 infer.py \
+    --checkpoint_path $CKPT_PATH \
+    --image_path $IMAGE_PATH \
+    --processor_path $FUDOKI_PATH \
+    --text_embedding_path $FUDOKI_PATH/text_embedding.pt \
+    --image_embedding_path $FUDOKI_PATH/image_embedding.pt \
+    --discrete_fm_steps 2 \
+    --seed 123

script/sft_debug.sh

@@ -0,0 +1,17 @@
+#!/bin/bash
+
+NUM_NODES=1
+NUM_GPUS=1
+
+config=config/debug.yaml
+output_dir=output/train/debug
+
+accelerate launch \
+  --config_file ./config/accelerate_config_ds2.yaml \
+  --machine_rank 0 \
+  --main_process_port 12345 \
+  --num_machines $NUM_NODES \
+  --num_processes $NUM_GPUS \
+  train.py \
+  --config $config \
+  --output_dir $output_dir

script/sft_navsim.sh

@@ -0,0 +1,17 @@
+#!/bin/bash
+
+NUM_NODES=1
+NUM_GPUS=1
+
+config=config/sft_navsim.yaml
+output_dir=output/train/debug
+
+accelerate launch \
+  --config_file ./config/accelerate_config_ds2.yaml \
+  --machine_rank 0 \
+  --main_process_port 12345 \
+  --num_machines $NUM_NODES \
+  --num_processes $NUM_GPUS \
+  train.py \
+  --config $config \
+  --output_dir $output_dir