A help request about training Diffusion Planner V4.0

[mcudsm]

HI
I've been trying to train Diffusion Planner V4.0, but the results haven't been satisfactory despite using around 100k npz files.My question is:
1. Could you please share how many npz files were used during your V4.0 training?
2. Also, if it's convenient, would it be possible for you to provide the .pth checkpoint? It would be incredibly helpful if I could fine-tune the model based on your pre-trained weights.

Thanks for your great work.

[SakodaShintaro]

The weights for Diffusion Planner v4.0 were trained using approximately 2M data.

The original Diffusion Planner paper also uses 1M data with nuPlan.

I will consult with the person in charge regarding whether the pth can be made public.

[mcudsm]

ok,Thanks!

[SakodaShintaro]

Weights for v5.0 are ready.

https://awf.ml.dev.web.auto/planning/models/diffusion_planner/v5.0/diffusion_planner.pth

[mcudsm]

Great ,Thanks a lot.