[quote="Lyken17, post:1, topic:11304"] ``` bwd_mod = relay.transform.gradient(mod['main'], mode="first_order") ``` [/quote]
Thanks for the information! This is exactly what I need. --- [Visit Topic](https://discuss.tvm.apache.org/t/errors-when-obtaining-gradients-for-nn-batch-norm/11304/3) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/6cd16e2e7fc091f5ff2d62a078c7a1cd6909bca0e2bd42bc705cdfe5eec3d665).
