autodl-projects/lib/nas_rnn/__init__.py

10 lines
285 B
Python
Raw Normal View History

2019-01-31 15:27:38 +01:00
# utils
from .utils import batchify, get_batch, repackage_hidden
# models
from .model_search import RNNModelSearch
from .model_search import DARTSCellSearch
from .basemodel import DARTSCell, RNNModel
# architecture
from .genotypes import DARTS_V1, DARTS_V2
from .genotypes import GDAS