Invalid shape during AutoTVM task creation in VTA tutorial

Hello All

I am trying to run the Auto-Tuning Tutorial (Link). I am getting this error “Invalid shape during AutoTVM task creation”.

How do I fix this?

Thanks in Advance !!