Downloads · 30 days
0
SARNN-MujocoUR5eCable
RoboManipBaselines/SARNN-MujocoUR5eCable
SARNN-MujocoUR5eCable is a machine learning model from RoboManipBaselines. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as bsd-2-clause.
The model was trained with the MujocoUR5eCable dataset.
Access
Public
Updated Apr 4, 2026
Repo size
7.3 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pkl2.2 MB · 84%
At a glance
- License
- bsd-2-clause
- Access
- Public
- Created
- Feb 12, 2025
- Updated
- Apr 4, 2026
- SHA
- a0bb6c9c
Model card
From the Hugging Face model README
Spatial attention recurrent neural network (SARNN)
The model was trained with the MujocoUR5eCable dataset.
Install
See GitHub for installation.
Policy rollout
Run a trained policy:
# Go to the top directory of this repository
$ cd robo_manip_baselines
$ python ./bin/Rollout.py Sarnn MujocoUR5eCable --checkpoint ./checkpoint/Sarnn/<checkpoint_name>/policy_last.ckpt
Technical Details
For more information on the technical details, please see the following paper:
@INPROCEEDINGS{SARNN_ICRA2022,
author = {Ichiwara, Hideyuki and Ito, Hiroshi and Yamamoto, Kenjiro and Mori, Hiroki and Ogata, Tetsuya},
title = {Contact-Rich Manipulation of a Flexible Object based on Deep Predictive Learning using Vision and Tactility},
booktitle = {International Conference on Robotics and Automation},
year = {2022},
pages = {5375-5381},
doi = {10.1109/ICRA46639.2022.9811940}
}
RoboManipBaselines Paper
- License
- bsd-2-clause
- Created
- Feb 12, 2025
- Updated
- Apr 4, 2026