talmolab / VNL-Brax-Imitation Public

Notifications You must be signed in to change notification settings
Fork 1
Star 0

VNL Implementation of Rodent Imitation Learning

0 stars 1 fork Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
assets		assets
.gitignore		.gitignore
README.md		README.md
environments.py		environments.py
losses.py		losses.py
mocap_preprocess.py		mocap_preprocess.py
networks.py		networks.py
obs_util.py		obs_util.py
ppo.py		ppo.py
process_traj.ipynb		process_traj.ipynb
train.py		train.py
walker.py		walker.py

Repository files navigation

Brax-Imitation

rodent imitation learning using brax

setup:

jax
brax
mujoco-mjx
flax
optax
dmcontrol
h5py
hydra-core

About

VNL Implementation of Rodent Imitation Learning

Custom properties

Report repository

Releases

No releases published

Packages

No packages published

Contributors 2

Languages