Refine lib -> xautodl

2021-05-19 14:17:20 +08:00
parent 4000f882f7
commit bda202ce87
4 changed files with 11 additions and 9 deletions
--- a/xautodl/models/shape_infers/InferMobileNetV2.py
+++ b/xautodl/models/shape_infers/InferMobileNetV2.py
@@ -2,7 +2,9 @@
 # Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019.01 #
 #####################################################
 # MobileNetV2: Inverted Residuals and Linear Bottlenecks, CVPR 2018
+#####################################################
 from torch import nn
+
 from ..initialization import initialize_resnet
 from ..SharedUtils import parse_channel_info