Commit 225986cb authored by YaningDylan's avatar YaningDylan
Browse files

minor

parent 858826a5

vagen/env/__init__.py

+1 −1
Original line number Diff line number Diff line 
from .sokoban import SokobanEnv,SokobanConfig

from .frozenlake import FrozenLakeEnv,FrozenLakeConfig

from .navigation import NavigationEnv, NavigationConfig

from .svg import SVGDINOEnv, SVGDINOConfig

from .svgdino import SVGDINOEnv, SVGDINOConfig

from .svg import SVGEnv, SVGConfig



REGISTERED_ENV = {

vagen/env/svgdino/env.py

+2 −2
Original line number Diff line number Diff line 
from vagen.env.base_env import BaseEnv

from vagen.env.svg.svg_utils import (process_and_rasterize_svg, is_valid_svg, load_svg_dataset)

from vagen.env.svg.score import calculate_total_score

from vagen.env.svgdino.svg_utils import (process_and_rasterize_svg, is_valid_svg, load_svg_dataset)

from vagen.env.svgdino.score import calculate_total_score

from vagen.env.utils.context_utils import parse_llm_raw_response, convert_numpy_to_PIL

from .config import SVGDINOConfig

from .prompt import system_prompt, init_observation_template, action_template

vagen/env/svgdino/score.py

+1 −1
Original line number Diff line number Diff line 
import numpy as np

import cv2

import os

from vagen.env.svg.dino import DINOScoreCalculator

from vagen.env.svgdino.dino import DINOScoreCalculator





def calculate_structural_accuracy(gt_im, gen_im):

vagen/examples/debug_svg_vision_grpo/run.sh

+1 −1
Original line number Diff line number Diff line
@@ -58,7 +58,7 @@ python3 -m vagen.trainer.main_ppo \ 
    trainer.critic_warmup=0 \

    trainer.logger=['console','wandb'] \

    trainer.project_name='vagen_debug' \

    trainer.experiment_name='grpo_mask_loss_structure_svg_vision_debug' \

    trainer.experiment_name='grpo_mask_loss_svg_vision_debug' \

    trainer.n_gpus_per_node=2 \

    trainer.nnodes=1 \

    trainer.save_freq=100 \

vagen/examples/debug_svgdino_vision_grpo/env_config.yaml

+2 −2
Original line number Diff line number Diff line 
env1:

    env_name: svg

    env_name: svgdino

    env_config:

        split: train

        dino_only: True
@@ -7,7 +7,7 @@ env1: 
    test_size: 0



env2:

    env_name: svg

    env_name: svgdino

    env_config:

        split: test

        dino_only: True

CRA Git | Maintained and supported by SUSTech CRA and CCSE