VLAarchtests3 / code /VLAarchtests2_code /reports /anybimanual_subset3_overlap_smoke200_train.log
lsnu's picture
Add files using upload-large-folder tool
31ade1f verified
/workspace/third_party/AnyBimanual/train.py:18: UserWarning:
The version_base parameter is not specified.
Please specify a compatability version level, or None.
Will assume defaults for version 1.1
@hydra.main(config_name="config", config_path="conf")
/workspace/envs/rlbench/lib/python3.10/site-packages/hydra/_internal/defaults_list.py:251: UserWarning: In 'config': Defaults list is missing `_self_`. See https://hydra.cc/docs/1.2/upgrades/1.0_to_1.1/default_composition_order for more information
warnings.warn(msg, UserWarning)
/workspace/envs/rlbench/lib/python3.10/site-packages/hydra/core/default_element.py:124: UserWarning: In 'method/PERACT_BC': Usage of deprecated keyword in package header '# @package _group_'.
See https://hydra.cc/docs/1.2/upgrades/1.0_to_1.1/changes_to_package_header for more information
deprecation_warning(
/workspace/envs/rlbench/lib/python3.10/site-packages/hydra/_internal/hydra.py:119: UserWarning: Future Hydra versions will no longer change working directory at job runtime by default.
See https://hydra.cc/docs/1.2/upgrades/1.1_to_1.2/changes_to_job_working_dir/ for more information.
ret = run_job(
[2026-03-30 05:02:50,133][root][INFO] -
method:
name: PERACT_BC
agent_type: leader_follower
robot_name: bimanual
image_crop_size: 64
bounds_offset:
- 0.15
voxel_sizes:
- 100
include_prev_layer: false
num_latents: 2048
latent_dim: 512
transformer_depth: 6
transformer_iterations: 1
cross_heads: 1
cross_dim_head: 64
latent_heads: 8
latent_dim_head: 64
pos_encoding_with_lang: true
conv_downsample: true
lang_fusion_type: seq
voxel_patch_size: 5
voxel_patch_stride: 5
final_dim: 64
low_dim_size: 4
input_dropout: 0.1
attn_dropout: 0.1
decoder_dropout: 0.0
lr: 0.0005
lr_scheduler: false
num_warmup_steps: 3000
optimizer: lamb
lambda_weight_l2: 1.0e-06
trans_loss_weight: 1.0
rot_loss_weight: 1.0
grip_loss_weight: 1.0
collision_loss_weight: 1.0
rotation_resolution: 5
activation: lrelu
norm: None
crop_augmentation: true
transform_augmentation:
apply_se3: true
aug_xyz:
- 0.125
- 0.125
- 0.125
aug_rpy:
- 0.0
- 0.0
- 45.0
aug_rot_resolution: ${method.rotation_resolution}
demo_augmentation: true
demo_augmentation_every_n: 10
no_skip_connection: false
no_perceiver: false
no_language: false
keypoint_method: heuristic
ddp:
master_addr: localhost
master_port: 12992
num_devices: 1
rlbench:
task_name: peract_bc_subset3_overlap_smoke200
tasks:
- coordinated_push_box
- coordinated_lift_ball
- dual_push_buttons
demos: 100
demo_path: /workspace/baselines/AnyBimanual_subset3_demo_root
episode_length: 25
cameras:
- over_shoulder_right
- wrist_right
- front
- overhead
- over_shoulder_left
- wrist_left
camera_resolution:
- 256
- 256
scene_bounds:
- -0.3
- -0.5
- 0.6
- 0.7
- 0.5
- 1.6
include_lang_goal_in_obs: true
instructions: /path
replay:
batch_size: 2
timesteps: 1
prioritisation: false
task_uniform: true
use_disk: true
path: /workspace/baselines/AnyBimanual_overlap_replay
max_parallel_processes: 8
task_folder: multi
framework:
log_freq: 50
save_freq: 200
train_envs: 1
replay_ratio: ${replay.batch_size}
transitions_before_train: 200
tensorboard_logging: true
csv_logging: true
training_iterations: 200
gpu: 0
env_gpu: 0
logdir: /workspace/baselines/AnyBimanual_overlap_runs
logging_level: 20
seeds: 1
start_seed: 0
load_existing_weights: false
num_weights_to_keep: 60
num_workers: 0
record_every_n: 5
checkpoint_name_prefix: checkpoint
use_wandb: false
wandb_project: ''
wandb_group: peract_bc_subset3_overlap_smoke200
seed: 0
wandb_name: peract_bc_subset3_overlap_smoke200
use_skill: true
use_pretrained: false
use_prefix: false
frozen: false
anybimanual: true
augmentation_type: ab
[2026-03-30 05:02:50,135][root][INFO] - CWD:/workspace/baselines/AnyBimanual_overlap_runs/peract_bc_subset3_overlap_smoke200/PERACT_BC
[2026-03-30 05:02:50,142][root][INFO] - Starting seed 0.
/workspace/third_party/AnyBimanual/train.py:18: UserWarning:
The version_base parameter is not specified.
Please specify a compatability version level, or None.
Will assume defaults for version 1.1
@hydra.main(config_name="config", config_path="conf")
[03/30/26 05:03:02] INFO INFO:root:Using method agent_factory.py:23
PERACT_BC with type
leader_follower
/workspace/baselines/AnyBimanual_overlap_replay/multi
No replay files found. Creating replay...
[03/30/26 05:03:05] INFO INFO:root:Creating a uniform_replay_buffer.py:156
TaskUniformReplayBuffe
r replay memory with
the following
parameters:
INFO INFO:root: uniform_replay_buffer.py:159
timesteps: 1
INFO INFO:root: uniform_replay_buffer.py:160
replay_capacity:
300000
INFO INFO:root: uniform_replay_buffer.py:161
batch_size: 2
INFO INFO:root: uniform_replay_buffer.py:162
update_horizon: 1
INFO INFO:root: uniform_replay_buffer.py:163
gamma: 0.990000
INFO INFO:root: uniform_replay_buffer.py:169
saving to disk:
/workspace/baselines/A
nyBimanual_overlap_rep
lay/multi
/workspace/third_party/AnyBimanual/train.py:18: UserWarning:
The version_base parameter is not specified.
Please specify a compatability version level, or None.
Will assume defaults for version 1.1
@hydra.main(config_name="config", config_path="conf")
/workspace/third_party/AnyBimanual/train.py:18: UserWarning:
The version_base parameter is not specified.
Please specify a compatability version level, or None.
Will assume defaults for version 1.1
@hydra.main(config_name="config", config_path="conf")
/workspace/third_party/AnyBimanual/train.py:18: UserWarning:
The version_base parameter is not specified.
Please specify a compatability version level, or None.
Will assume defaults for version 1.1
@hydra.main(config_name="config", config_path="conf")
/workspace/third_party/AnyBimanual/train.py:18: UserWarning:
The version_base parameter is not specified.
Please specify a compatability version level, or None.
Will assume defaults for version 1.1
@hydra.main(config_name="config", config_path="conf")
Found 4 keypoints. [50, 82, 84, 95]
Found 4 keypoints. [49, 95, 97, 118]
Found 4 keypoints. [65, 85, 87, 103]
Found 4 keypoints. [48, 86, 88, 98]
Found 4 keypoints. [63, 100, 102, 120]
Found 4 keypoints. [60, 86, 88, 100]
Found 4 keypoints. [47, 94, 96, 108]
Found 4 keypoints. [48, 88, 90, 100]
Found 4 keypoints. [48, 90, 92, 103]
Found 4 keypoints. [60, 88, 90, 102]
Found 4 keypoints. [49, 94, 96, 108]
Found 4 keypoints. [48, 85, 87, 97]
Found 4 keypoints. [50, 78, 80, 91]
Found 4 keypoints. [48, 87, 89, 100]
Found 4 keypoints. [53, 86, 88, 99]
Found 4 keypoints. [47, 96, 98, 113]
Found 4 keypoints. [53, 92, 94, 109]
Found 4 keypoints. [62, 87, 89, 103]
Found 4 keypoints. [52, 86, 88, 99]
Found 4 keypoints. [61, 99, 101, 122]
Found 4 keypoints. [56, 88, 90, 101]
Found 4 keypoints. [58, 93, 95, 108]
Found 4 keypoints. [67, 85, 87, 106]
Found 4 keypoints. [48, 91, 93, 105]
Found 4 keypoints. [67, 91, 93, 109]
Found 4 keypoints. [53, 85, 87, 98]
Found 4 keypoints. [49, 96, 98, 110]
Found 4 keypoints. [50, 86, 88, 99]
Found 4 keypoints. [49, 83, 85, 96]
Found 4 keypoints. [51, 95, 97, 110]
Found 4 keypoints. [51, 97, 99, 118]
Found 4 keypoints. [64, 77, 79, 95]
Found 4 keypoints. [49, 82, 84, 94]
Found 4 keypoints. [49, 93, 95, 106]
Found 4 keypoints. [56, 98, 100, 118]
Found 4 keypoints. [62, 93, 95, 110]
Found 4 keypoints. [56, 91, 93, 105]
Found 4 keypoints. [55, 87, 89, 100]
Found 4 keypoints. [54, 81, 83, 102]
Found 4 keypoints. [65, 94, 96, 112]
Found 4 keypoints. [68, 100, 102, 164]
Found 4 keypoints. [47, 81, 83, 93]
Found 4 keypoints. [56, 88, 90, 102]
Found 4 keypoints. [67, 96, 98, 118]
Found 4 keypoints. [61, 88, 90, 103]
Found 4 keypoints. [59, 93, 95, 109]
Found 4 keypoints. [47, 87, 89, 99]
Found 4 keypoints. [67, 87, 89, 107]
Found 4 keypoints. [50, 87, 89, 99]
Found 4 keypoints. [68, 82, 84, 104]
Found 4 keypoints. [49, 94, 96, 108]
Found 4 keypoints. [57, 88, 90, 101]
Found 4 keypoints. [64, 73, 75, 91]
Found 4 keypoints. [50, 84, 86, 96]
Found 4 keypoints. [65, 89, 91, 108]
Found 4 keypoints. [48, 83, 85, 95]
Found 4 keypoints. [66, 76, 78, 93]
Found 4 keypoints. [56, 81, 83, 95]
Found 4 keypoints. [60, 80, 82, 95]
Found 4 keypoints. [66, 96, 98, 121]
Found 4 keypoints. [61, 87, 89, 102]
Found 4 keypoints. [66, 94, 96, 115]
Found 4 keypoints. [65, 85, 87, 101]
Found 4 keypoints. [64, 98, 100, 115]
Found 4 keypoints. [57, 91, 93, 105]
Found 4 keypoints. [50, 86, 88, 98]
Found 4 keypoints. [48, 83, 85, 95]
Found 4 keypoints. [51, 77, 79, 90]
Found 4 keypoints. [55, 83, 85, 97]
Found 4 keypoints. [47, 89, 91, 103]
Found 4 keypoints. [48, 99, 101, 116]
Found 4 keypoints. [52, 83, 85, 95]
Found 4 keypoints. [56, 97, 99, 122]
Found 4 keypoints. [47, 102, 104, 130]
Found 4 keypoints. [47, 83, 85, 95]
Found 4 keypoints. [48, 89, 91, 103]
Found 4 keypoints. [50, 84, 86, 97]
Found 4 keypoints. [55, 91, 93, 105]
Found 4 keypoints. [63, 90, 92, 106]
Found 4 keypoints. [59, 87, 89, 101]
Found 4 keypoints. [68, 72, 74, 96]
Found 4 keypoints. [68, 86, 88, 108]
Found 4 keypoints. [64, 86, 88, 103]
Found 4 keypoints. [65, 82, 84, 100]
Found 4 keypoints. [54, 87, 89, 101]
Found 4 keypoints. [55, 95, 97, 109]
Found 4 keypoints. [52, 90, 92, 105]
Found 4 keypoints. [57, 90, 92, 105]
Found 4 keypoints. [66, 93, 95, 112]
Found 4 keypoints. [48, 92, 94, 105]
Found 4 keypoints. [48, 81, 83, 93]
Found 4 keypoints. [54, 84, 86, 97]
Found 5 keypoints. [48, 78, 80, 94, 96]
Found 4 keypoints. [66, 91, 93, 111]
Found 4 keypoints. [66, 88, 90, 107]
Found 4 keypoints. [54, 99, 101, 121]
Found 4 keypoints. [54, 89, 91, 102]
Found 4 keypoints. [51, 88, 90, 101]
Found 4 keypoints. [61, 94, 96, 112]
Found 4 keypoints. [47, 84, 86, 97]
Found 2 keypoints. [101, 157]
Found 1 keypoints. [131]
Found 2 keypoints. [94, 146]
Found 2 keypoints. [81, 131]
Found 2 keypoints. [84, 136]
Found 2 keypoints. [93, 141]
Found 2 keypoints. [92, 143]
Found 2 keypoints. [94, 149]
Found 2 keypoints. [86, 135]
Found 2 keypoints. [99, 153]
Found 2 keypoints. [96, 147]
Found 2 keypoints. [102, 154]
Found 2 keypoints. [86, 134]
Found 2 keypoints. [83, 136]
Found 2 keypoints. [86, 134]
Found 2 keypoints. [87, 135]
Found 2 keypoints. [96, 147]
Found 2 keypoints. [88, 135]
Found 2 keypoints. [86, 136]
Found 2 keypoints. [86, 135]
Found 2 keypoints. [85, 136]
Found 2 keypoints. [85, 134]
Found 2 keypoints. [92, 143]
Found 2 keypoints. [96, 149]
Found 2 keypoints. [100, 157]
Found 2 keypoints. [93, 142]
Found 2 keypoints. [84, 137]
Found 2 keypoints. [97, 154]
Found 2 keypoints. [86, 134]
Found 2 keypoints. [91, 138]
Found 2 keypoints. [95, 146]
Found 2 keypoints. [86, 135]
Found 2 keypoints. [85, 135]
Found 2 keypoints. [94, 150]
Found 2 keypoints. [89, 136]
Found 2 keypoints. [89, 139]
Found 2 keypoints. [95, 151]
Found 2 keypoints. [90, 137]
Found 2 keypoints. [103, 155]
Found 2 keypoints. [89, 140]
Found 2 keypoints. [85, 134]
Found 2 keypoints. [91, 139]
Found 2 keypoints. [89, 139]
Found 2 keypoints. [87, 137]
Found 2 keypoints. [104, 155]
Found 2 keypoints. [91, 141]
Found 2 keypoints. [87, 138]
Found 2 keypoints. [96, 148]
Found 2 keypoints. [91, 138]
Found 2 keypoints. [104, 162]
Found 2 keypoints. [87, 136]
Found 2 keypoints. [88, 134]
Found 2 keypoints. [87, 135]
Found 2 keypoints. [86, 135]
Found 2 keypoints. [90, 135]
Found 2 keypoints. [86, 135]
Found 2 keypoints. [86, 135]
Found 2 keypoints. [76, 129]
Found 2 keypoints. [87, 137]
Found 2 keypoints. [90, 139]
Found 2 keypoints. [88, 139]
Found 2 keypoints. [94, 148]
Found 2 keypoints. [104, 154]
Found 2 keypoints. [92, 139]
Found 2 keypoints. [99, 153]
Found 2 keypoints. [96, 151]
Found 2 keypoints. [87, 137]
Found 2 keypoints. [92, 146]
Found 2 keypoints. [78, 125]
Found 2 keypoints. [86, 136]
Found 2 keypoints. [78, 132]
Found 2 keypoints. [97, 150]
Found 2 keypoints. [89, 135]
Found 2 keypoints. [83, 132]
Found 2 keypoints. [85, 134]
Found 2 keypoints. [91, 141]
Found 2 keypoints. [87, 135]
Found 2 keypoints. [91, 145]
Found 2 keypoints. [92, 141]
Found 2 keypoints. [93, 142]
Found 2 keypoints. [88, 134]
Found 2 keypoints. [87, 135]
Found 2 keypoints. [92, 138]
Found 1 keypoints. [127]
Found 2 keypoints. [88, 139]
Found 2 keypoints. [93, 144]
Found 2 keypoints. [85, 136]
Found 2 keypoints. [85, 135]
Found 2 keypoints. [84, 137]
Found 2 keypoints. [92, 142]
Found 2 keypoints. [85, 136]
Found 2 keypoints. [93, 143]
Found 2 keypoints. [81, 129]
Found 2 keypoints. [84, 136]
Found 2 keypoints. [85, 135]
Found 2 keypoints. [89, 139]
Found 2 keypoints. [90, 142]
Found 2 keypoints. [86, 137]
Found 2 keypoints. [82, 130]
Found 2 keypoints. [93, 146]
Found 4 keypoints. [73, 89, 97, 129]
Found 4 keypoints. [76, 92, 100, 125]
Found 4 keypoints. [67, 83, 90, 116]
Found 4 keypoints. [65, 81, 89, 120]
Found 4 keypoints. [79, 112, 120, 162]
Found 4 keypoints. [66, 83, 91, 119]
Found 4 keypoints. [80, 96, 104, 131]
Found 4 keypoints. [78, 97, 105, 142]
Found 4 keypoints. [72, 89, 97, 132]
Found 4 keypoints. [76, 93, 102, 129]
Found 4 keypoints. [68, 84, 92, 116]
Found 4 keypoints. [70, 86, 94, 119]
Found 4 keypoints. [73, 91, 98, 134]
Found 4 keypoints. [73, 88, 96, 126]
Found 4 keypoints. [84, 103, 111, 164]
Found 4 keypoints. [80, 98, 106, 143]
Found 4 keypoints. [77, 94, 102, 147]
Found 4 keypoints. [73, 93, 102, 133]
Found 4 keypoints. [70, 87, 95, 121]
Found 4 keypoints. [72, 90, 99, 129]
Found 4 keypoints. [68, 85, 94, 121]
Found 4 keypoints. [82, 99, 106, 145]
Found 4 keypoints. [74, 90, 98, 128]
Found 4 keypoints. [74, 91, 99, 136]
Found 4 keypoints. [65, 83, 92, 122]
Found 4 keypoints. [65, 80, 89, 117]
Found 4 keypoints. [78, 96, 105, 140]
Found 4 keypoints. [80, 99, 106, 156]
Found 4 keypoints. [68, 84, 92, 120]
Found 4 keypoints. [76, 93, 102, 129]
Found 4 keypoints. [80, 102, 109, 137]
Found 4 keypoints. [79, 103, 110, 155]
Found 4 keypoints. [78, 95, 104, 148]
Found 4 keypoints. [74, 91, 99, 138]
Found 4 keypoints. [74, 91, 99, 131]
Found 4 keypoints. [82, 99, 108, 145]
Found 4 keypoints. [70, 86, 94, 125]
Found 4 keypoints. [66, 82, 90, 120]
Found 4 keypoints. [65, 83, 92, 118]
Found 4 keypoints. [73, 97, 104, 141]
Found 4 keypoints. [78, 106, 114, 145]
Found 4 keypoints. [72, 87, 95, 118]
Found 7 keypoints. [63, 66, 73, 74, 99, 107, 143]
Found 4 keypoints. [76, 93, 100, 141]
Found 4 keypoints. [72, 88, 95, 134]
Found 4 keypoints. [86, 103, 111, 148]
Found 4 keypoints. [78, 100, 108, 135]
Found 4 keypoints. [70, 86, 94, 125]
Found 4 keypoints. [76, 92, 99, 138]
Found 4 keypoints. [69, 85, 93, 125]
Found 4 keypoints. [69, 85, 93, 126]
Found 4 keypoints. [78, 95, 102, 140]
Found 4 keypoints. [69, 86, 94, 122]
Found 4 keypoints. [78, 99, 109, 144]
Found 4 keypoints. [68, 89, 98, 123]
Found 4 keypoints. [77, 95, 102, 129]
Found 4 keypoints. [79, 99, 108, 135]
Found 4 keypoints. [82, 99, 108, 140]
Found 4 keypoints. [68, 85, 93, 119]
Found 4 keypoints. [74, 90, 98, 133]
Found 4 keypoints. [82, 99, 108, 143]
Found 4 keypoints. [79, 95, 103, 135]
Found 4 keypoints. [75, 89, 97, 130]
Found 4 keypoints. [77, 94, 103, 139]
Found 4 keypoints. [73, 91, 97, 142]
Found 4 keypoints. [76, 92, 101, 131]
Found 4 keypoints. [76, 93, 102, 130]
Found 4 keypoints. [80, 98, 107, 148]
Found 4 keypoints. [79, 105, 113, 146]
Found 4 keypoints. [67, 84, 93, 121]
Found 4 keypoints. [76, 93, 102, 137]
Found 4 keypoints. [74, 90, 98, 128]
Found 4 keypoints. [73, 89, 97, 133]
Found 4 keypoints. [77, 94, 102, 140]
Found 4 keypoints. [72, 95, 102, 141]
Found 4 keypoints. [67, 83, 91, 125]
Found 4 keypoints. [73, 89, 98, 123]
Found 4 keypoints. [76, 92, 100, 136]
Found 4 keypoints. [78, 96, 105, 139]
Found 4 keypoints. [79, 95, 103, 137]
Found 4 keypoints. [70, 86, 95, 119]
Found 4 keypoints. [68, 84, 91, 126]
Found 4 keypoints. [67, 85, 94, 119]
Found 4 keypoints. [66, 82, 90, 118]
Found 4 keypoints. [75, 92, 100, 124]
Found 4 keypoints. [71, 88, 96, 128]
Found 4 keypoints. [67, 82, 91, 118]
Found 4 keypoints. [81, 101, 110, 145]
Found 4 keypoints. [79, 95, 103, 139]
Found 4 keypoints. [70, 111, 120, 150]
Found 4 keypoints. [65, 82, 91, 118]
Found 4 keypoints. [74, 91, 99, 143]
Found 4 keypoints. [79, 112, 121, 148]
Found 4 keypoints. [81, 97, 104, 136]
Found 4 keypoints. [75, 92, 100, 136]
Found 4 keypoints. [73, 88, 96, 121]
Found 4 keypoints. [72, 89, 98, 123]
Found 4 keypoints. [77, 94, 103, 129]
Found 4 keypoints. [83, 101, 110, 147]
Found 4 keypoints. [64, 83, 91, 117]
[03/30/26 07:29:49] INFO INFO:root:# Q qattention_peract_bc_agent.py:315
Params: 44677231
INFO INFO:root:# Q qattention_peract_bc_agent.py:315
Params: 44677743
Error executing job with overrides: ['method=PERACT_BC', 'ddp.num_devices=1', 'ddp.master_port=12992', 'framework.logdir=/workspace/baselines/AnyBimanual_overlap_runs', 'framework.start_seed=0', 'framework.use_wandb=False', 'framework.wandb_project=', 'framework.wandb_group=peract_bc_subset3_overlap_smoke200', 'framework.wandb_name=peract_bc_subset3_overlap_smoke200', 'framework.gpu=0', 'framework.env_gpu=0', 'framework.anybimanual=True', 'framework.augmentation_type=ab', 'framework.training_iterations=200', 'framework.save_freq=200', 'framework.log_freq=50', 'framework.seeds=1', 'framework.num_workers=0', 'replay.batch_size=2', 'replay.path=/workspace/baselines/AnyBimanual_overlap_replay', 'replay.task_folder=multi', 'replay.max_parallel_processes=8', 'rlbench.task_name=peract_bc_subset3_overlap_smoke200', 'rlbench.tasks=[coordinated_push_box,coordinated_lift_ball,dual_push_buttons]', 'rlbench.demo_path=/workspace/baselines/AnyBimanual_subset3_demo_root', 'rlbench.demos=100', 'rlbench.episode_length=25']
Traceback (most recent call last):
File "/workspace/third_party/AnyBimanual/train.py", line 98, in main
mp.spawn(
File "/workspace/envs/rlbench/lib/python3.10/site-packages/torch/multiprocessing/spawn.py", line 340, in spawn
return start_processes(fn, args, nprocs, join, daemon, start_method="spawn")
File "/workspace/envs/rlbench/lib/python3.10/site-packages/torch/multiprocessing/spawn.py", line 296, in start_processes
while not context.join():
File "/workspace/envs/rlbench/lib/python3.10/site-packages/torch/multiprocessing/spawn.py", line 211, in join
raise ProcessRaisedException(msg, error_index, failed_process.pid)
torch.multiprocessing.spawn.ProcessRaisedException:
-- Process 0 terminated with the following error:
Traceback (most recent call last):
File "/workspace/envs/rlbench/lib/python3.10/site-packages/torch/multiprocessing/spawn.py", line 87, in _wrap
fn(i, *args)
File "/workspace/third_party/AnyBimanual/run_seed_fn.py", line 238, in run_seed
train_runner.start()
File "/workspace/third_party/AnyBimanual/third_party/YARR/yarr/runners/offline_train_runner.py", line 125, in start
self._agent.load_weights(file_path)
File "/workspace/third_party/AnyBimanual/third_party/YARR/yarr/agents/agent.py", line 337, in load_weights
self.leader_agent.load_weights(savedir)
File "/workspace/third_party/AnyBimanual/helpers/preprocess_agent.py", line 125, in load_weights
self._pose_agent.load_weights(savedir)
File "/workspace/third_party/AnyBimanual/agents/peract_bc/qattention_stack_agent.py", line 128, in load_weights
qa.load_weights(savedir)
File "/workspace/third_party/AnyBimanual/agents/peract_bc/qattention_peract_bc_agent.py", line 861, in load_weights
state_dict = torch.load(weight_file, map_location=device)
File "/workspace/envs/rlbench/lib/python3.10/site-packages/torch/serialization.py", line 1530, in load
with _open_file_like(f, "rb") as opened_file:
File "/workspace/envs/rlbench/lib/python3.10/site-packages/torch/serialization.py", line 795, in _open_file_like
return _open_file(name_or_buffer, mode)
File "/workspace/envs/rlbench/lib/python3.10/site-packages/torch/serialization.py", line 776, in __init__
super().__init__(open(name, mode)) # noqa: SIM115
FileNotFoundError: [Errno 2] No such file or directory: '/path/to/your/pre-trained ckpts/PERACT_BC/checkpoint_peract_bc_leader_layer_0.pt'
Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.