iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refine lib -> xautodl

Browse Source
This commit is contained in:
D-X-Y 2021-05-19 14:17:20 +08:00
parent 4000f882f7
commit bda202ce87
4 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_math_adv.py
View File

@ -4,14 +4,12 @@
# pytest tests/test_math_adv.py -s # # pytest tests/test_math_adv.py -s #
##################################################### #####################################################
import unittest import unittest
import pytest
from xautodl import datasets from xautodl.datasets.math_core import QuadraticFunc
from datasets.math_core import QuadraticFunc from xautodl.datasets.math_core import ConstantFunc
from datasets.math_core import ConstantFunc from xautodl.datasets.math_core import DynamicLinearFunc
from datasets.math_core import DynamicLinearFunc from xautodl.datasets.math_core import DynamicQuadraticFunc
from datasets.math_core import DynamicQuadraticFunc from xautodl.datasets.math_core import ComposedSinFunc
from datasets.math_core import ComposedSinFunc
class TestConstantFunc(unittest.TestCase): class TestConstantFunc(unittest.TestCase):

1
xautodl/models/shape_infers/InferCifarResNet.py
View File

@ -3,6 +3,7 @@
##################################################### #####################################################
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from ..initialization import initialize_resnet from ..initialization import initialize_resnet

2
xautodl/models/shape_infers/InferMobileNetV2.py
View File

@ -2,7 +2,9 @@
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019.01 # # Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019.01 #
##################################################### #####################################################
# MobileNetV2: Inverted Residuals and Linear Bottlenecks, CVPR 2018 # MobileNetV2: Inverted Residuals and Linear Bottlenecks, CVPR 2018
#####################################################
from torch import nn from torch import nn
from ..initialization import initialize_resnet from ..initialization import initialize_resnet
from ..SharedUtils import parse_channel_info from ..SharedUtils import parse_channel_info

5
xautodl/models/shape_infers/InferTinyCellNet.py
View File

@ -3,8 +3,9 @@
##################################################### #####################################################
from typing import List, Text, Any from typing import List, Text, Any
import torch.nn as nn 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): class DynamicShapeTinyNet(nn.Module):