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
|
84c733fa85
|
fix: remove prefetcher, put normalizer in model
1. remove messy data prefetcher which will cause confusion
2. put normliazer in model to accelerate training via GPU computing
|
2020-05-25 23:39:11 +08:00 |
|
liaoxingyu
|
651e6ba9c4
|
feat: support multiprocess predictor
add asyncpredictor to support multiprocessing feature extraction with dataloader
|
2020-05-09 18:23:36 +08:00 |
|