https://github.com/sisl/mechamodlearn
Structured framework for learning mechanical systems in PyTorch
https://github.com/sisl/mechamodlearn
lagrangian machine-learning model-learning physics pytorch robotics
Last synced: 9 months ago
JSON representation
Structured framework for learning mechanical systems in PyTorch
- Host: GitHub
- URL: https://github.com/sisl/mechamodlearn
- Owner: sisl
- License: mit
- Created: 2019-02-19T18:19:15.000Z (over 7 years ago)
- Default Branch: master
- Last Pushed: 2019-04-15T21:46:49.000Z (about 7 years ago)
- Last Synced: 2024-08-04T03:51:53.984Z (almost 2 years ago)
- Topics: lagrangian, machine-learning, model-learning, physics, pytorch, robotics
- Language: Python
- Homepage: https://arxiv.org/abs/1902.08705
- Size: 36.1 KB
- Stars: 23
- Watchers: 19
- Forks: 5
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# MechaModLearn
Authors: [@kunalmenda](https://github.com/kunalmenda) and [@rejuvyesh](https://github.com/rejuvyesh)
This library provides a structured framework for learning mechanical systems in PyTorch.
---
## Installation
Requires Python3.
```
git clone https://github.com/sisl/mechamodlearn.git
cd mechamodlearn
pip install -e .
```
## Usage
Example experiments are placed in [`experiments`](./experiments) directory.
To run the Simple experiment:
```
python experiments/simple.py --logdir /tmp/simplelog
```
## References
---
If you found this library useful in your research, please consider citing our [paper](https://arxiv.org/abs/1902.08705):
```
@article{gupta2019mod,
title={A General Framework for Structured Learning of Mechanical Systems},
author={Gupta, Jayesh K. and Menda, Kunal and Manchester, Zachary and Kochenderfer, Mykel},
journal={arXiv preprint arXiv:1902.08705},
year={2019}
}
```