search

oke-aditya / image_similarity

PyTorch Blog Post On Image Similarity Search
https://oke-aditya.github.io/image_similarity/
Apache License 2.0
249 stars 59 forks source link

Update torch_engine.py from "loss = loss_fn(enc_output, dec_output" to "loss = loss_fn(dec_output, target_img)" #3

Closed JonathanSum closed 4 years ago

oke-aditya commented 4 years ago

Absolutely right @JonathanSum . I missed this thing in val_step. It is correct in train_step. Thnx for this !!

oke-aditya commented 4 years ago

The reason for such bugs was because I trained on Kaggle, where I used the ipynb file and not the scripts. Such stuff fixed on fly might not be there in scripts. Thnx for that !!

JonathanSum commented 4 years ago

sato love