Commit Graph

12 Commits (5bd26d8aa27756b040b5661c084ded41e828b7a9)

Author SHA1 Message Date
Ramakrishnan Sundareswaran 32acbbe4ad
Update mudeep.py to support testing
Current model fails in the case of triplet loss during testing since y,x returns a tuple where engine.py features expects a torch tensor instead to run features.cpu()
2021-04-08 11:31:27 -05:00
KaiyangZhou 08bb2cc5eb linting and formatting code 2019-12-02 14:34:59 +00:00
KaiyangZhou 6b8fe56638 linting and formatting code 2019-12-01 02:35:44 +00:00
KaiyangZhou a136384a6a rm unused imports/variables; fixed minor bugs 2019-10-23 17:15:37 +01:00
KaiyangZhou 4ca182dbb8 add model public keys 2019-03-24 15:52:39 +00:00
KaiyangZhou aa98ef5a7b add docstrings to losses, metrics, models, utils, optim 2019-03-22 00:14:41 +00:00
KaiyangZhou 37f92b7cc3 create engine SDK 2019-03-19 17:26:08 +00:00
KaiyangZhou 03d0674dcb comply woth PEP8 2019-03-15 23:17:38 +00:00
KaiyangZhou 564f5a26ab update print 2019-01-30 22:41:47 +00:00
KaiyangZhou 4ce9024fcc add reference to all models 2018-10-27 17:46:48 +01:00
KaiyangZhou d076e41e78 add def featuremaps(self, x) 2018-10-27 10:49:27 +01:00
KaiyangZhou 611ac80f0b mv deepreid/ to torchreid/ 2018-08-15 09:48:17 +01:00