Aditya Oke

Hi! most probably the dimensions of your image are not same. You would need to resize them. This is due to assumption int the tutorial that all the images have...

Have you written in the `torch_train.py` code. If yes then it should work fine. ``` loss_fn = nn.MSELoss() ```

I'm still perplexed. It should run as I had tested this code (long time ago) and it ran fine. Even the ipynb has the output. https://github.com/oke-aditya/image_similarity/blob/master/image_similarity/train-image-similarity.ipynb

Can you share a reproducible script over colab so that I can debug?

Yes of coure you can retrain these models on any dataset. The training scripts are provided here. Note that this similarity search works on unlabelled data. It can help to...

Hi @hassiahk Thanks for the issue. I guess this can be better. Currently, I'm not maintaining/developing this repo any more. I know that this project was quite good and people...

Hi @hassiahk !! Happy to say that this entire code and much more is shifted to complete CV Package under work [Here](https://github.com/Quick-AI/quickvision) is the link to it. It has super...

You can take this up surely. It's up for grabs ! LR finder is a standard technique whose implementation I have pointed out. Most code is going to be edit...

Yes, I guess you get the point. But let me clarify what exactly can be done. It has a class for LR finder on [this](https://github.com/davidtvs/pytorch-lr-finder/blob/master/torch_lr_finder/lr_finder.py#L107) line. I want that to...

I would stick to PyTorch. This library has to work with torch. 😀 Give a try lets see if we get bugs.