Commit Graph

29 Commits (3b4f5f4dfcabd3abc143d5859d54ce3ec8be5d10)

Author SHA1 Message Date
littletomatodonkey 3b4f5f4dfc
add distillation and fix some apis (#810)
* fix save load and imagenet dataset
* refine trainer
2021-06-10 16:30:05 +08:00
Felix 47039c826f
Update trainer.py 2021-06-08 19:25:00 +08:00
littletomatodonkey b9044c270a add standard logger 2021-06-08 05:21:53 +00:00
weishengyu 934de96539 rename cam -> unique 2021-06-05 23:06:33 +08:00
weishengyu 9e4a1045f0 dbg 2021-06-05 16:17:14 +08:00
weishengyu 81c864b96d dbg 2021-06-05 16:12:35 +08:00
weishengyu cd7f606a5f dbg 2021-06-05 16:02:22 +08:00
weishengyu 2a72c57e92 dbg remove same image 2021-06-05 15:56:41 +08:00
weishengyu 5597830e4c dbg camera 2021-06-05 15:35:20 +08:00
weishengyu 19be351646 dbg 2021-06-05 15:12:38 +08:00
weishengyu 0a8b1773ba dbg 2021-06-05 14:10:20 +08:00
weishengyu 2a0333281f dbg 2021-06-05 13:53:41 +08:00
weishengyu 1c7a23c5e7 dbg 2021-06-05 13:44:58 +08:00
weishengyu f9b4e0f0bb dbg 2021-06-05 11:36:52 +08:00
weishengyu d954f73888 remove to_tensor 2021-06-04 23:40:46 +08:00
weishengyu b48d2c7a97 dbg 2021-06-04 23:12:07 +08:00
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 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