DiTree-RHC: model checkpoints

Flow-matching policy checkpoints for DiTree-RHC, DiTree extended to a quadrotor in 3D voxel mazes as a global planner and as a receding-horizon local planner.

Code: https://github.com/YuvalDara/DiTree-RHC

file	description
dronemaze_05_10_13_10/dronemaze_05_10_13_10_epoch_120_step_5760000.pt	model used for all reported results (120 epochs)
dronemaze_04_09_19_26/dronemaze_04_09_19_26_epoch_60_final.pt	earlier model (60 epochs)

Both were trained from scratch on the dataset in datasets/Yuval535/DiTree-RHC. Each file is a full training checkpoint (network weights, EMA weights, optimizer and scheduler state).

Download into the code repository:

hf download Yuval535/DiTree-RHC --include "*.pt" --local-dir ditree/checkpoints

Built on DiTree (Hassidof, Jurgenson, Solovey, CoRL 2025).