Commit Graph

104 Commits (ff5f983c7f2147e72f65440a94e7c4087aad214a)

Author SHA1 Message Date
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