wayveai / mile

PyTorch code for the paper "Model-Based Imitation Learning for Urban Driving".
MIT License
330 stars 31 forks source link

Shu.ishida/adding leaderboard eval #11

Closed ishida-wayve closed 1 year ago

ishida-wayve commented 1 year ago

Enabling models to be initialised with the proper pretrained weight paths (rather than using the path stored in the checkpoint), as well as adding in an agent that extends from AutonomousAgent class that can be submitted to the leaderboard.