Results 3 comments of yjy

Thank you very much。 I add a forward method in LitResNetTransformer just like you said。 but fail another way ```python def load_model(): lit_model = LitResNetTransformer.load_from_checkpoint("artifacts/model_basic_2.ckpt") lit_model.freeze() # lit_model.eval() x =...

could you give a export onnx example in this project。 https://pytorch-lightning.readthedocs.io/en/latest/common/production_inference.html