Refine lib -> xautodl

xautodl/models/shape_infers/InferCifarResNet.py

@@ -3,6 +3,7 @@
 #####################################################
 import torch.nn as nn
 import torch.nn.functional as F
+
 from ..initialization import initialize_resnet
 
 

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
 

xautodl/models/shape_infers/InferTinyCellNet.py

@@ -3,8 +3,9 @@
 #####################################################
 from typing import List, Text, Any
 import torch.nn as nn
-from models.cell_operations import ResNetBasicblock
-from models.cell_infers.cells import InferCell
+
+from ..cell_operations import ResNetBasicblock
+from ..cell_infers.cells import InferCell
 
 
 class DynamicShapeTinyNet(nn.Module):