@MarisaKirisame Do you mean you want the training loop itself to be part of Relay?
-- 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-538140726
