Just to clarify: > There is two function, loss and infer, which one calculate loss and gradient, > and one just does infer.
Do you mean that there are two modes, training and inference, so that there are multiple entry points in relay Module? -- You are receiving this because you are subscribed to this thread. Reply to this email directly or view it on GitHub: https://github.com/dmlc/tvm/issues/4054#issuecomment-538058861
