Brax Rodent Run - Online Training Script

Environment Setup

Name		Name	Last commit message	Last commit date
Latest commit History 68 Commits
models		models
preprocessing		preprocessing
.gitignore		.gitignore
Env_step.ipynb		Env_step.ipynb
Rodent_Env_Brax.py		Rodent_Env_Brax.py
brax_rodent_run_ppo.py		brax_rodent_run_ppo.py
custom_losses.py		custom_losses.py
custom_networks.py		custom_networks.py
custom_ppo.py		custom_ppo.py
custom_ppo_networks.py		custom_ppo_networks.py
custom_wrappers.py		custom_wrappers.py
env_testing.ipynb		env_testing.ipynb
new_slurm.py		new_slurm.py
readme.md		readme.md
render_rollout.ipynb		render_rollout.ipynb
slurm.py		slurm.py