forked from tangger/lerobot
ff82367c62
- Introduced Ensemble and CriticHead classes for more efficient critic network handling - Added support for multiple camera inputs in observation encoder - Optimized image encoding by batching image processing - Updated configuration for ManiSkill environment with reduced image size and action scaling - Compiled critic networks for improved performance - Simplified normalization and ensemble handling in critic networks Co-authored-by: michel-aractingi <michel.aractingi@gmail.com>
20 lines
317 B
YAML
20 lines
317 B
YAML
# @package _global_
|
|
|
|
fps: 20
|
|
|
|
env:
|
|
name: maniskill/pushcube
|
|
task: PushCube-v1
|
|
image_size: 64
|
|
control_mode: pd_ee_delta_pose
|
|
state_dim: 25
|
|
action_dim: 7
|
|
fps: ${fps}
|
|
obs: rgb
|
|
render_mode: rgb_array
|
|
render_size: 64
|
|
device: cuda
|
|
|
|
reward_classifier:
|
|
pretrained_path: null
|
|
config_path: null |