Fix evaluate.py and put dataset sizes inside params

omoindrot / tensorflow-triplet-loss

Implementation of triplet loss in TensorFlow

https://omoindrot.github.io/triplet-loss

MIT License

1.12k stars 283 forks source link

Closed georgymh closed 6 years ago

georgymh commented 6 years ago

Great repository and blog post!

This pull request changes 2 things:

Fixes evaluate.py since it wasn't running because of the data input_fn
Addresses the TODO about putting the dataset sizes inside the parameters files

omoindrot commented 6 years ago

Hi @georgymh, thanks for the pull request I had already done some of the modifications on my side but didn't push it to the repo.

I added a few things you did and pushed it myself so I'll close this pull request ! Thanks :)