Commit e2100fb8 authored by jameskrw's avatar jameskrw
Browse files

minor

parent cf20367c

vagen/env/primitive_skill/maniskill/env/place_two_cube.py

+1 −1
Original line number Diff line number Diff line
@@ -95,7 +95,7 @@ class PlaceTwoCubeEnv(BaseEnv): 
            # Place cubes at random positions

            xyz = torch.zeros((b, 3))

            xyz[:, 2] = 0.02

            region = [[self.workspace_x[0], self.workspace_y[0]], [self.workspace_x[1], self.workspace_y[1]]]

            region = [[self.workspace_x[0], self.workspace_y[0]], [self.workspace_x[1], -0.02]] 

            sampler = randomization.UniformPlacementSampler(bounds=region, batch_size=b)

            radius = torch.linalg.norm(torch.tensor([0.02, 0.02])) + 0.02

vagen/examples/debug/debug_primitive_skill_text_grpo_service/env_config.yaml

+1 −0
Original line number Diff line number Diff line
@@ -2,5 +2,6 @@ env1: 
    env_name: primitive_skill 

    env_config:

        render_mode: text

        env_id: "PlaceTwoCube" # AlignTwoCube,PlaceTwoCube,PutAppleInDrawer,StackThreeCube

    train_size: 10000  

    test_size: 32
 
 No newline at end of file

vagen/examples/debug/debug_primitive_skill_text_grpo_service/run.sh

+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/navigation-vision-debug/train.parquet" \

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

    --force_gen



# max_trajectory_length = max_prompt_length + max_response_length

CRA Git | Maintained and supported by SUSTech CRA and CCSE