Allow checkpoints to be exported to ONNX after transferring between machines
Currently in to_onnx
the model is loaded directly, rather than going through the CLI. This means that the norm_config
parameters are not correctly set. We should either offload stuff from the CLI into the model files themselves, or load the model using the CLI in the to_onnx
script
possibly related to https://github.com/Lightning-AI/pytorch-lightning/pull/18105
cc @npond