Commit 90358cd5 authored by root's avatar root
Browse files

update script

parent ddf5e5de

vagen/env/svg/env.py

+0 −1
Original line number Diff line number Diff line
@@ -137,7 +137,6 @@ class SVGEnv(BaseEnv): 
                self.gen_image = gen_image

                # Calculate score

                score_config = self.config.get_score_config()

                print(f"score_config:{score_config}")

                scores = calculate_total_score(

                    gt_im=self.gt_image,

                    gen_im=gen_image,

vagen/examples/svg_aico/env_config.yaml

+1 −1
Original line number Diff line number Diff line
@@ -4,7 +4,7 @@ env1: 
        split: train

        dino_weight: 1.0

        structural_weight: 1.0

    train_size: 10000

    train_size: 1000

    test_size: 0



env2:

vagen/examples/svg_aico/run.sh

100644 → 100755
+1 −0
Original line number Diff line number Diff line
@@ -10,6 +10,7 @@ python -m vagen.env.create_dataset \ 
    --yaml_path "$SCRIPT_DIR/env_config.yaml" \

    --train_path "data/svg-vision-debug/train.parquet" \

    --test_path "data/svg-vision-debug/test.parquet" \

    --force_gen



# max_trajectory_length = max_prompt_length + max_response_length

vagen/examples/svg_trico/env_config.yaml

+1 −1
Original line number Diff line number Diff line
@@ -4,7 +4,7 @@ env1: 
        split: train

        dino_weight: 1.0

        structural_weight: 1.0

    train_size: 10000

    train_size: 1000

    test_size: 0



env2:

vagen/examples/svg_trico/run.sh

+1 −1
Original line number Diff line number Diff line
@@ -21,7 +21,7 @@ python3 -m vagen.trainer.main_ppo \ 
    data.train_batch_size=16 \

    data.max_prompt_length=1024 \

    data.max_response_length=648 \

    data.max_trajectory_length=3600 \

    data.max_trajectory_length=1800 \

    data.image_key=images \

    data.truncation=error \

    actor_rollout_ref.model.path=Qwen/Qwen2.5-VL-3B-Instruct \

CRA Git | Maintained and supported by SUSTech CRA and CCSE