Refine lib -> xautodl
This commit is contained in:
parent
4000f882f7
commit
bda202ce87
@ -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):
|
||||
|
@ -3,6 +3,7 @@
|
||||
#####################################################
|
||||
import torch.nn as nn
|
||||
import torch.nn.functional as F
|
||||
|
||||
from ..initialization import initialize_resnet
|
||||
|
||||
|
||||
|
@ -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
|
||||
|
||||
|
@ -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):
|
||||
|
Loading…
Reference in New Issue
Block a user