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