Buggy SSD implementation

I was trying to perform object detection as SSD using fastai_v1 by modifying @rohitgeo’s code here:
https://colab.research.google.com/drive/10zfoO8xYPXUsUVh_SVeGlzFTkEOCcBJX#scrollTo=IFeN9-uiceJh

The NB starts giving me these weird error messages the moment I start running learn.lr_find() :




RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM

Can anyone please help?