MarkFzp / act-plus-plus

Imitation learning algorithms with Co-training for Mobile ALOHA: ACT, Diffusion Policy, VINN
https://mobile-aloha.github.io/
MIT License
2.86k stars 525 forks source link

RuntimeError: torch.cat(): expected a non-empty list of Tensors #43

Open liqingxin opened 4 months ago

liqingxin commented 4 months ago

I met an issue while training the model.

I have already run the episodes in simulation 屏幕截图 2024-03-19 10:45:34

but Found 0 hdf5 files

Data from: ['~/aloha/act-plus-plus-main/data/sim_transfer_cube_scripted']

Traceback (most recent call last): File "imitate_episodes.py", line 667, in main(vars(parser.parse_args())) File "imitate_episodes.py", line 166, in main train_dataloader, valdataloader, stats, = load_data(dataset_dir, name_filter, camera_names, batch_size_train, batch_size_val, args['chunk_size'], args['skip_mirrored_data'], config['load_pretrain'], policy_class, stats_dir_l=stats_dir, sample_weights=sample_weights, train_ratio=train_ratio) File "/home/lqx/aloha/act-plus-plus-main/utils.py", line 245, in loaddata , all_episode_len = get_norm_stats(dataset_path_list) File "/home/lqx/aloha/act-plus-plus-main/utils.py", line 174, in get_norm_stats all_qpos_data = torch.cat(all_qpos_data, dim=0) RuntimeError: torch.cat(): expected a non-empty list of Tensors

eric-hu519 commented 4 months ago

same question.Do u solve this problem?I m confused.

kimkimyoung commented 3 months ago

modify the data root in constant.py

sainavaneet commented 3 months ago

modify the data root in constant.py

what do you mean by that ?

kimkimyoung commented 3 months ago

modify the data root in constant.py

what do you mean by that ?

in constant.py

# DATA_DIR = '/home/zfu/interbotix_ws/src/act/data' if os.getlogin() == 'zfu' else '/scr/tonyzhao/datasets'
DATA_DIR = '/home/kim/PCNL-Works/act-plus-plus/dataset'
sainavaneet commented 3 months ago

modify the data root in constant.py

what do you mean by that ?

in constant.py

# DATA_DIR = '/home/zfu/interbotix_ws/src/act/data' if os.getlogin() == 'zfu' else '/scr/tonyzhao/datasets'
DATA_DIR = '/home/kim/PCNL-Works/act-plus-plus/dataset'

thank you got it