Brax-Imitation

Rodent imitation learning using Brax and MJX

Current Progress

Intention Network + Expert overlay rendering.

Name		Name	Last commit message	Last commit date
Latest commit History 11 Commits
assets		assets
configs		configs
envs		envs
ppo_imitation		ppo_imitation
rendering		rendering
.gitignore		.gitignore
README.md		README.md
ant_traj.p		ant_traj.p
ant_traj_still.p		ant_traj_still.p
brax_ppo.py		brax_ppo.py
environment.yml		environment.yml
environments_explore.ipynb		environments_explore.ipynb
humanoid_traj.p		humanoid_traj.p
humanoid_traj_stand.p		humanoid_traj_stand.p
mocap_preprocess.py		mocap_preprocess.py
networks.py		networks.py
process_traj.ipynb		process_traj.ipynb
render_expert.ipynb		render_expert.ipynb
requirements.txt		requirements.txt
train.py		train.py
train_expert.ipynb		train_expert.ipynb
transform_snips_250_clip_750.p		transform_snips_250_clip_750.p
transform_snips_groom.p		transform_snips_groom.p
walker.py		walker.py