omron-sinicx / neural-astar

Official implementation of "Path Planning using Neural A* Search" (ICML-21)
https://omron-sinicx.github.io/neural-astar
Other
236 stars 57 forks source link

Introduce PTL #4

Closed yonetaniryo closed 1 year ago

yonetaniryo commented 1 year ago
yonetaniryo commented 1 year ago

Comparison between original (orange) / PTL training (blue)

image

*difference in train_loss is due to the change from epoch-based to batch-based stats