Triplet Correlation Detector Inference

Thank you for the awesome code. I have trained the triplet detector for cars on kitti and I am able to get visualization of filter outputs. But unfortunately I am unable to run the tester.py. There is issue with mismatch in saved model state dict.
I was wondering if you can inform how to get inference of the correlation detector on an entire image (like the ones showed in homepage)?
