Refine lib -> xautodl

2021-05-19 13:00:33 +08:00
parent 94a149b33f
commit 95fb5a54b1
34 changed files with 118 additions and 1154 deletions
--- a/xautodl/xlayers/super_trade_stem.py
+++ b/xautodl/xlayers/super_trade_stem.py
@@ -12,7 +12,7 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F

-import spaces
+from xautodl import spaces
 from .super_linear import SuperLinear
 from .super_module import SuperModule
 from .super_module import IntSpaceType