GitLab

* Add loss, sampler, and train script

* Fix train script

* Add argparse

* Fix lint

* Change f strings to .format()

* Remove unused imports

* Change TripletMarginLoss to extend nn.Module

* Load eye uint8 tensors directly on device

* Refactor model.py to backbone=None

* Add docstring for PKSampler

* Refactor evaluate() to take loader as arg instead

* Change eval method to cat embeddings all at once

* Add dataset comments

* Add README.md

* Add tests for sampler

* Refactor threshold finder to helper method

* Refactor targets comment

* Fix lint

* Rename embedding to similarity (More consistent with existing literature)