Skip to content

v0.3.0

Compare
Choose a tag to compare
@fedebotu fedebotu released this 10 Nov 06:33
· 673 commits to main since this release

Faster Library, Python 3.11 and new TorchRL support, Envs, Models, Multiple Dataloaders, and more 🚀

Faster Library, new Python 3.11 and TorchRL

  • Update to latest TorchRL #72, solving several issues as #95 #97 (also see this)
  • Benchmarking:
    • Up to 20% speedup in training epochs thanks to faster TensorDict and new env updates
    • Almost instant data generation (avoid list comprehension, e.g. from ~20 seconds to <1 second per epoch!)
  • Python 3.11 now available #97

New SMTWTP environment

  • Add new scheduling problem: Single Machine Total Weighted Tardiness Problem environment as in DeepACO @henry-yeh

New MatNet model

  • Add MatNet version for square matrices (faster implementation ideal for routing problems)
  • Should be easy to implement scheduling from here

Multiple Dataloaders

  • Now it is possible to have multiple dataloaders, with naming as well!
    • For example, to track generalization during training

Miscellaneous

  • Fix POMO shapes @hyeok9855 , modularizing PPO etc
  • Fix precion bug for PPO
  • New AI4CO transfer!