Limited torch.compile support
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