omron-sinicx/neural-astar

Introduce PTL

Closed this issue · 1 comments

yonetaniryo commented
  • Update train.py to use PTL functions
  • Update PyTorch version
yonetaniryo commented

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

image

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