fix: some cfg issue (RLinf#302)

Signed-off-by: hongzhi <zanghongzhi@infini-ai.com>

Author: zanghz21

examples/embodiment/config/env/eval/PutOnPlateInScene25Main.yaml

@@ -17,7 +17,7 @@ video_cfg:
 init_params:
   id: PutOnPlateInScene25Main-v3
   num_envs: 4
-  obs_mode: ${env.eval.num_envs}
+  obs_mode: ${env.train.init_params.obs_mode}
   control_mode: None
   sim_backend: "gpu"
   sim_config: ${env.train.init_params.sim_config}

examples/embodiment/config/maniskill_grpo_openvla.yaml

@@ -148,7 +148,7 @@ actor:
     attn_implementation: "flash_attention_2"
     low_cpu_mem_usage: True
     trust_remote_code: True
-    gradient_checkpointing: False
+    gradient_checkpointing: True
 
   optim:
     lr: 1.0e-5

examples/embodiment/config/maniskill_grpo_openvlaoft.yaml

@@ -140,7 +140,7 @@ actor:
     attn_implementation: "flash_attention_2"
     low_cpu_mem_usage: True
     trust_remote_code: True
-    gradient_checkpointing: False
+    gradient_checkpointing: True
 
   tokenizer:
     tokenizer_type: "HuggingFaceTokenizer"

examples/embodiment/config/maniskill_ppo_openvla.yaml

@@ -139,7 +139,7 @@ actor:
     attn_implementation: "flash_attention_2"
     low_cpu_mem_usage: True
     trust_remote_code: True
-    gradient_checkpointing: False
+    gradient_checkpointing: True
 
   optim:
     lr: 1.0e-4

examples/embodiment/config/maniskill_ppo_openvla_eval.yaml

@@ -140,7 +140,7 @@ actor:
     attn_implementation: "flash_attention_2"
     low_cpu_mem_usage: True
     trust_remote_code: True
-    gradient_checkpointing: False
+    gradient_checkpointing: True
 
   optim:
     lr: 1.0e-4