Refine lib -> xautodl

This commit is contained in:
D-X-Y
2021-05-19 13:00:33 +08:00
parent 94a149b33f
commit 95fb5a54b1
34 changed files with 118 additions and 1154 deletions

View File

@@ -8,7 +8,7 @@ import torch.nn.functional as F
import math
from typing import Optional, Callable, Tuple
import spaces
from xautodl import spaces
from .super_module import SuperModule
from .super_module import IntSpaceType
from .super_module import BoolSpaceType