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 07:19:20 +00:00
parent bda202ce87
commit 5b9a028e60
19 changed files with 46 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
xautodl/models/cell_infers/cells.py
View File

@@ -6,7 +6,7 @@ import torch
import torch.nn as nn
from copy import deepcopy
from models.cell_operations import OPS
from xautodl.models.cell_operations import OPS
# Cell for NAS-Bench-201

1
xautodl/models/cell_infers/nasnet_cifar.py
View File

@@ -4,6 +4,7 @@
import torch
import torch.nn as nn
from copy import deepcopy
from .cells import NASNetInferCell as InferCell, AuxiliaryHeadCIFAR

10
xautodl/models/xcore.py
View File

@@ -9,11 +9,11 @@ import torch
__all__ = ["get_model"]
from xlayers.super_core import SuperSequential
from xlayers.super_core import SuperLinear
from xlayers.super_core import SuperDropout
from xlayers.super_core import super_name2norm
from xlayers.super_core import super_name2activation
from xautodl.xlayers.super_core import SuperSequential
from xautodl.xlayers.super_core import SuperLinear
from xautodl.xlayers.super_core import SuperDropout
from xautodl.xlayers.super_core import super_name2norm
from xautodl.xlayers.super_core import super_name2activation
def get_model(config: Dict[Text, Any], **kwargs):