search

woven-planet / l5kit

L5Kit - https://woven.toyota
https://woven-planet.github.io/l5kit
858 stars 277 forks source link

PPO policy and custom feature extractor added #324

Closed theDebugger811 closed 3 years ago

theDebugger811 commented 3 years ago

The PR comprises:

  1. A simple CNN backbone feature extractor with group norm (GN)
  2. Custom Feature extractor for PPO using CNN+GN
  3. Callbacks to save and evaluate model state
  4. A script to run PPO on L5Kit.
theDebugger811 commented 3 years ago

@perone , @lucabergamini and @szagoruyko could you please review the PR which incorporates the PPO Policy into L5Kit? Thanks!

theDebugger811 commented 3 years ago

Thanks @perone for the quick review. I have incorporated the suggestions. Could you have another pass?

theDebugger811 commented 3 years ago

Thanks @perone , comments incorporated.