AlexanderKoch-Koch / low_cost_robot

MIT License
3k stars 255 forks source link

AI training #13

Closed vateseif closed 5 months ago

vateseif commented 5 months ago

I've re-adapted the implementation from ACT to work on low_cost_robot.

record_episodes.py: stores teleoperation episodes in .hdf5 format train.py: trains the policies and saves checkpoints in checkpoints/ eval.py: evaluates the policy on the robot

Happy to do modifications if there's any feedback/request.