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/InferTinyCellNet.py
+++ b/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):