Skip to content

Limited torch.compile support

Samuel Van Stroud requested to merge svanstro/salt:svanstro/compile-model into main

Add a --compile flag to the CLI which compiles the full lightning model. Since torch.compile does not yet support dynamic shapes, the classification task was modified to have a fixed shape (see !128 (merged)), and the vertexing task is not supported.

See #19

Edited by Samuel Van Stroud

Merge request reports