search

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.98k stars 551 forks source link

TypeError: forward() get an unexpected keyword argument 'pos' #5

Closed Ke-Wang1017 closed 9 months ago

Ke-Wang1017 commented 9 months ago

managed to load data but has error as the title Screenshot from 2024-01-04 11-12-50

apirrone commented 9 months ago

I have the exact same issue. I'm digging into the code and will report back here if I find something :)

apirrone commented 9 months ago

Doing this https://github.com/MarkFzp/act-plus-plus/issues/3#issuecomment-1877316563 seems to solve the problem

Ke-Wang1017 commented 9 months ago

yep as mentioned by @apirrone , in detr/models/detr_vae.py at line 285 change build_transformer to build_encoder will solve the issue and mine starts to train now