import compute_model_complexity

2019-05-22 16:14:47 +01:00 · 2019-05-22 16:14:47 +01:00 · 26cf6be779
parent e0a23bab07
commit 26cf6be779
1 changed files with 2 additions and 1 deletions
--- a/torchreid/utils/init.py
+++ b/torchreid/utils/init.py
@ -5,4 +5,5 @@ from .loggers import *
 from .tools import *
 from .reidtools import *
 from .torchtools import *
-from .rerank import re_ranking
+from .rerank import re_ranking
+from .model_complexity import compute_model_complexity