Commit Graph

14 Commits (57ebf75f834b28d9a4442b6060a834c9856e74bc)

Author SHA1 Message Date
liaoxingyu 65169b40bd support ddp testing 2020-07-30 20:15:28 +08:00
liaoxingyu 64861ea6c8 update config file 2020-07-29 19:48:49 +08:00
liaoxingyu a8c7e07df6 update deploy docs 2020-07-29 17:47:35 +08:00
liaoxingyu 16655448c2 onnx/trt support
Summary: change model pretrain mode and support onnx/TensorRT export
2020-07-29 17:43:39 +08:00
liaoxingyu fec7abc461 finish v0.2 ddp training 2020-07-06 16:57:43 +08:00
liaoxingyu 8879db3fba update training instruction
Summary: update dataset configuration and training instruction
2020-06-16 19:43:36 +08:00
liaoxingyu 27c48c8f02 update pytorchtocaffe tool
Summary: update pytorch to caffe toolchain to support model layers and bash scripts
2020-06-09 14:45:27 +08:00
Christian Clauss d7cf32a652
Fix a Python syntax error discovered by #77 2020-06-03 14:32:46 +02:00
liaoxingyu 69e4ba901e update acknowledgement in deploy 2020-06-02 18:12:57 +08:00
liaoxingyu 2f6d999469 add pytorch to caffe converting
Summary: update deployment README and support pytorch to caffe converting for basemodel
2020-06-02 18:11:02 +08:00
liaoxingyu 4be4cacb73 fix: add a simple way to reset data prefetcher when resume training
use data prefetcher build-in reset function to reload it rather than
redefining a new data prefetcher, otherwise it will introduce other
problems in eval-only mode.
2020-05-09 11:58:27 +08:00
liaoxingyu d27729c5bb refactor(preciseBN): add preciseBN datasets show 2020-04-29 21:05:53 +08:00
liaoxingyu ec19bcc1d3 style(configs): put all config files together
put all config files into one place for easily control,
and add tools for put train_net.py which almost the same in
different projects
2020-04-29 16:18:54 +08:00
liaoxingyu e38a799b63 fix(engine/defaults): fix precise bn bug
fix problem in precise bn, which will not use precise bn datasets, and throw some errors
2020-04-29 16:16:54 +08:00