Refine lib -> xautodl

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

View File

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

View File

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

View File

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

View File

@ -3,8 +3,9 @@
#####################################################
from typing import List, Text, Any
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):