search

facebookresearch / r3m

Pre-training Reusable Representations for Robotic Manipulation Using Diverse Human Video Data
https://sites.google.com/view/robot-r3m/
MIT License
292 stars 45 forks source link

Access to the language&vision module pre-trained weights #30

Open KevinRiou22 opened 1 year ago

KevinRiou22 commented 1 year ago

Hi, Can we have access to the weights of the trained language&vision module (pre-trained weights with "lang_enc" keys), so that I can use the pre-trained R3M with the language head (removing the call to "remove_language_head()" function in the load_r3m() function). Thanks,