liaoxingyu 16655448c2 onnx/trt support ...

Summary: change model pretrain mode and support onnx/TensorRT export

2020-07-29 17:43:39 +08:00

..

__init__.py

1. upload circle loss and arcface

2020-04-05 23:54:26 +08:00

autoaugment.py

support DDP sampler and augment

2020-07-06 16:55:23 +08:00

build.py

onnx/trt support

2020-07-29 17:43:39 +08:00

functional.py

update augmix augments

2020-05-27 22:56:31 +08:00

transforms.py

onnx/trt support

2020-07-29 17:43:39 +08:00