iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refine lib -> xautodl

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
xautodl/xlayers/super_activations.py
View File

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

2
xautodl/xlayers/super_attention.py
View File

@@ -13,7 +13,7 @@ import torch.nn as nn
import torch.nn.functional as F
import spaces
from xautodl import spaces
from .super_module import SuperModule
from .super_module import IntSpaceType
from .super_module import BoolSpaceType

2
xautodl/xlayers/super_container.py
View File

@@ -9,7 +9,7 @@ import operator
from collections import OrderedDict
from typing import Optional, Union, Callable, TypeVar, Iterator
import spaces
from xautodl import spaces
from .super_module import SuperModule

2
xautodl/xlayers/super_dropout.py
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

2
xautodl/xlayers/super_linear.py
View File

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

3
xautodl/xlayers/super_module.py
View File

@@ -12,8 +12,7 @@ import torch
import torch.nn as nn
from enum import Enum
import spaces
import xautodl.spaces
from .super_utils import IntSpaceType, BoolSpaceType
from .super_utils import LayerOrder, SuperRunMode
from .super_utils import TensorContainer

2
xautodl/xlayers/super_norm.py
View File

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

2
xautodl/xlayers/super_positional_embedding.py
View File

@@ -5,7 +5,7 @@ import torch
import torch.nn as nn
import math
import spaces
from xautodl import spaces
from .super_module import SuperModule
from .super_module import IntSpaceType

2
xautodl/xlayers/super_trade_stem.py
View File

@@ -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

2
xautodl/xlayers/super_transformer.py
View File

@@ -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_module import IntSpaceType
from .super_module import BoolSpaceType
from .super_module import LayerOrder

2
xautodl/xlayers/super_utils.py
View File

@@ -9,7 +9,7 @@ import torch
import torch.nn as nn
from enum import Enum
import spaces
from xautodl import spaces
IntSpaceType = Union[int, spaces.Integer, spaces.Categorical]
BoolSpaceType = Union[bool, spaces.Categorical]