New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Torch batchnorm issue with Aimet 1.32.2 #3152

Open

shayaanjamil-10xe opened this issue Jul 8, 2024 · 0 comments

shayaanjamil-10xe commented Jul 8, 2024 •

edited

Loading

Getting this error when I am trying to train RTM-det with QAT.

Steps I took to prepare for QAT:

My package versions
aimet-torch 1.32.2
torch 2.2.2
with cuda 12.0

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment