Commit Graph

215 Commits (3be87e0d91c3d9738eab89f9933e558aebbed22e)

Author SHA1 Message Date
weishengyu 3e4af0448e add default metrics 2021-06-04 22:56:12 +08:00
weishengyu b23e72b17c add combined_metrics 2021-06-04 22:27:32 +08:00
weishengyu b56237dc99 update trainer 2021-06-04 21:42:23 +08:00
weishengyu ed098b3c7d modify trainer 2021-06-04 21:11:01 +08:00
dongshuilong 3ef9993026 fix ReID trainer bugs 2021-06-03 20:14:45 +08:00
dongshuilong c784209355 fix ReID trainer 2021-06-03 19:24:50 +08:00
weishengyu 69f563d234 rename losses -> loss 2021-06-03 15:17:49 +08:00
dongshuilong 435144b526 fix bugs imgnet and rec train 2021-06-03 12:24:48 +08:00
dongshuilong 452f532121 add CompCars train and fix bugs 2021-06-03 11:00:47 +08:00
dongshuilong bba0cf8fdb add CompCars train 2021-06-02 20:04:24 +08:00
cuicheng01 d13cb22378 Add legendary_model's congigs and fix some trainer's bug 2021-06-01 17:24:21 +00:00
littletomatodonkey 0a0e34298b
add support for infer and export (#760)
* add support for infer

* add export model support

* fix yaml

* fix post process name

* fix topk name
2021-06-01 14:37:47 +08:00
littletomatodonkey 3881343484
add support for train and eval (#752)
* add support for train and eval

* rm unsed code

* add support for metric save and load ckp
2021-05-29 13:52:32 +08:00
littletomatodonkey dd79f81fd7
[WIP]add arch init (#744)
* polish trainer
2021-05-27 18:41:44 +08:00
weishengyu f6da3bd60a move files 2021-05-24 11:42:24 +08:00