diff --git a/README.md b/README.md index 9fafde3c..8fc7f9f8 100644 --- a/README.md +++ b/README.md @@ -155,6 +155,9 @@ python train.py -b [batch_size] -lr [learning rate] -e [epochs] - This repo is mainly inspired by [deepinsight/insightface](https://github.com/deepinsight/insightface) and [InsightFace_TF](https://github.com/auroua/InsightFace_TF) +## 5. [Tensorrt](https://github.com/jinbaoziyl/InsightFace_Tensorrtx.git) + + ## PS - PRs are welcome, in case that I don't have the resource to train some large models like the 100 and 151 layers model