fix log_alpha in modeling_sac: change to nn.parameter
added pretrained vision model in policy Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
This commit is contained in:
committed by
AdilZouitine
parent
57e09828ce
commit
a0e0a9a9b1
4
lerobot/configs/env/so100_real.yaml
vendored
4
lerobot/configs/env/so100_real.yaml
vendored
@@ -1,6 +1,6 @@
|
||||
# @package _global_
|
||||
|
||||
fps: 30
|
||||
fps: 10
|
||||
|
||||
env:
|
||||
name: real_world
|
||||
@@ -26,6 +26,6 @@ env:
|
||||
joint_masking_action_space: [1, 1, 1, 1, 0, 0] # disable wrist and gripper
|
||||
|
||||
reward_classifier:
|
||||
pretrained_path: outputs/classifier/checkpoints/best/pretrained_model
|
||||
pretrained_path: outputs/classifier/13-02-random-sample-resnet10-frozen/checkpoints/best/pretrained_model
|
||||
config_path: lerobot/configs/policy/hilserl_classifier.yaml
|
||||
|
||||
@@ -8,7 +8,7 @@
|
||||
# env.gym.obs_type=environment_state_agent_pos \
|
||||
|
||||
seed: 1
|
||||
dataset_repo_id: null # aractingi/push_green_cube_hf_cropped_resized
|
||||
dataset_repo_id: aractingi/push_cube_square_light_offline_demo_cropped_resized
|
||||
|
||||
training:
|
||||
# Offline training dataloader
|
||||
@@ -52,7 +52,7 @@ policy:
|
||||
n_action_steps: 1
|
||||
|
||||
shared_encoder: true
|
||||
# vision_encoder_name: null
|
||||
vision_encoder_name: "helper2424/resnet10"
|
||||
freeze_vision_encoder: true
|
||||
input_shapes:
|
||||
# # TODO(rcadene, alexander-soare): add variables for height and width from the dataset/env?
|
||||
|
||||
Reference in New Issue
Block a user