17 Commits

Author SHA1 Message Date
KaiyangZhou
36e22e8ce1 make engine more model-agnostic 2020-04-16 12:46:15 +01:00
KaiyangZhou
08bb2cc5eb linting and formatting code 2019-12-02 14:34:59 +00:00
KaiyangZhou
58e6841a33 update relative import 2019-12-01 03:31:32 +00:00
KaiyangZhou
6b8fe56638 linting and formatting code 2019-12-01 02:35:44 +00:00
KaiyangZhou
307e38e519 add use_unlabeled_targets 2019-11-27 16:35:54 +00:00
KaiyangZhou
5c9bac5e75 minor modification 2019-11-27 11:09:13 +00:00
KaiyangZhou
a136384a6a rm unused imports/variables; fixed minor bugs 2019-10-23 17:15:37 +01:00
KaiyangZhou
5365619606 add config 2019-08-26 10:34:31 +01:00
KaiyangZhou
92f1ea215a add tensorboard 2019-08-22 22:41:21 +01:00
KaiyangZhou
1558389f37 osnet.py: rename residual->identity; engine: add num_batches=len(dloader) 2019-08-21 23:02:05 +01:00
kaiyangzhou
f57d88decf move fixbase into train 2019-05-24 15:36:10 +01:00
KaiyangZhou
4b93121496 minor 2019-04-19 11:52:02 +01:00
KaiyangZhou
4a033659b0 do not decay lr while fixbase=True 2019-03-29 11:40:23 +00:00
KaiyangZhou
1e9d466f42 move scheduler.step() to train() 2019-03-27 13:01:39 +00:00
KaiyangZhou
d0fe37c558 finalize docs 2019-03-23 23:09:39 +00:00
KaiyangZhou
46a0d5c27d update dataset and docs 2019-03-21 12:53:21 +00:00
KaiyangZhou
37f92b7cc3 create engine SDK 2019-03-19 17:26:08 +00:00