Skip to content

Torch 2.1, model compilation

Samuel Van Stroud requested to merge svanstro/salt:svanstro/bump-versions into main
  • bump python version to 3.11
  • bump torch version to 2.1.1
  • torch.compile is working, 30% speed up
  • move type alias to avoid collisions with stdlib
  • add script to clean empty logdirs

TODO

  • test onnx export with new onnx version
Edited by Samuel Van Stroud

Merge request reports