test_touch_x / files /config.yaml
sach088's picture
Upload folder using huggingface_hub
78fce10 verified
_wandb:
value:
cli_version: 0.21.4
e:
0pemdxvajth4tsbx7fmf6a6nwx8h5zp9:
args:
- --dataset.repo_id=sach088/test_touch_x
- --policy.type=act
- --output_dir=outputs/train/test_touch_x
- --job_name=test_touch_x
- --wandb.enable=true
- --policy.repo_id=sach088/test_touch_x
codePath: src/lerobot/scripts/lerobot_train.py
codePathLocal: src/lerobot/scripts/lerobot_train.py
cpu_count: 1
cpu_count_logical: 2
cudaVersion: "12.4"
disk:
/:
total: "120942624768"
used: "49230315520"
executable: /usr/bin/python3
git:
commit: 3f8c5d9809d07e7767f2a1366fee86e6e27640a2
remote: https://github.com/huggingface/lerobot.git
gpu: Tesla T4
gpu_count: 1
gpu_nvidia:
- architecture: Turing
cudaCores: 2560
memoryTotal: "16106127360"
name: Tesla T4
uuid: GPU-1f8e3433-1ef1-7ed5-413f-ffb28672af0a
host: 6a07fdd2461a
memory:
total: "13605847040"
os: Linux-6.6.105+-x86_64-with-glibc2.35
program: /content/lerobot/src/lerobot/scripts/lerobot_train.py
python: CPython 3.12.12
root: outputs/train/test_touch_x
startedAt: "2025-10-30T16:50:06.940701Z"
writerId: 0pemdxvajth4tsbx7fmf6a6nwx8h5zp9
m: []
python_version: 3.12.12
t:
"1":
- 1
- 2
- 3
- 5
- 11
- 12
- 41
- 49
- 51
- 53
- 71
- 83
- 105
"2":
- 1
- 2
- 3
- 5
- 11
- 12
- 41
- 49
- 51
- 53
- 71
- 83
- 105
"3":
- 13
- 15
- 16
- 61
"4": 3.12.12
"5": 0.21.4
"6": 4.57.1
"10":
- 21
"12": 0.21.4
"13": linux-x86_64
batch_size:
value: 8
checkpoint_path:
value: null
dataset:
value:
episodes: null
image_transforms:
enable: false
max_num_transforms: 3
random_order: false
tfs:
affine:
kwargs:
degrees:
- -5
- 5
translate:
- 0.05
- 0.05
type: RandomAffine
weight: 1
brightness:
kwargs:
brightness:
- 0.8
- 1.2
type: ColorJitter
weight: 1
contrast:
kwargs:
contrast:
- 0.8
- 1.2
type: ColorJitter
weight: 1
hue:
kwargs:
hue:
- -0.05
- 0.05
type: ColorJitter
weight: 1
saturation:
kwargs:
saturation:
- 0.5
- 1.5
type: ColorJitter
weight: 1
sharpness:
kwargs:
sharpness:
- 0.5
- 1.5
type: SharpnessJitter
weight: 1
repo_id: sach088/test_touch_x
revision: null
root: null
streaming: false
use_imagenet_stats: true
video_backend: torchcodec
env:
value: null
eval:
value:
batch_size: 50
n_episodes: 50
use_async_envs: false
eval_freq:
value: 20000
job_name:
value: test_touch_x
log_freq:
value: 200
num_workers:
value: 4
optimizer:
value:
betas:
- 0.9
- 0.999
eps: 1e-08
grad_clip_norm: 10
lr: 1e-05
type: adamw
weight_decay: 0.0001
output_dir:
value: outputs/train/test_touch_x
policy:
value:
chunk_size: 100
device: cuda
dim_feedforward: 3200
dim_model: 512
dropout: 0.1
feedforward_activation: relu
kl_weight: 10
latent_dim: 32
license: null
n_action_steps: 100
n_decoder_layers: 1
n_encoder_layers: 4
n_heads: 8
n_obs_steps: 1
n_vae_encoder_layers: 4
normalization_mapping:
ACTION: MEAN_STD
STATE: MEAN_STD
VISUAL: MEAN_STD
optimizer_lr: 1e-05
optimizer_lr_backbone: 1e-05
optimizer_weight_decay: 0.0001
pre_norm: false
pretrained_backbone_weights: ResNet18_Weights.IMAGENET1K_V1
pretrained_path: null
private: null
push_to_hub: true
replace_final_stride_with_dilation: false
repo_id: sach088/test_touch_x
tags: null
temporal_ensemble_coeff: null
type: act
use_amp: false
use_vae: true
vision_backbone: resnet18
resume:
value: false
save_checkpoint:
value: true
save_freq:
value: 20000
scheduler:
value: null
seed:
value: 1000
steps:
value: 100000
use_policy_training_preset:
value: true
wandb:
value:
disable_artifact: false
enable: true
entity: null
mode: null
notes: null
project: lerobot
run_id: null