Commit 32695db6 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

Refactoring

parent cd88e2e2
......@@ -164,12 +164,13 @@ if CUDA:
from .nnet import StatDataset
from .nnet import Xtractor
from .nnet import xtrain
from .nnet import xtrain_single
from .nnet import xtrain_asynchronous
from .nnet import extract_idmap
from .nnet import extract_parallel
from .nnet import extract_embeddings
from .nnet import ResBlock
from .nnet import ResNet18
#from .nnet import SAD_RNN
else:
print("Don't import Torch")
......
......@@ -31,7 +31,7 @@ Copyright 2014-2020 Anthony Larcher and Sylvain Meignier
from .feed_forward import FForwardNetwork
from .feed_forward import kaldi_to_hdf5
from .xsets import XvectorMultiDataset, XvectorDataset, StatDataset
from .xvector import Xtractor, xtrain, extract_idmap, extract_parallel, xtrain_single
from .xvector import Xtractor, xtrain, extract_idmap, extract_parallel, xtrain_asynchronous, extract_embeddings
from .res_net import ResBlock, ResNet18
__author__ = "Anthony Larcher and Sylvain Meignier"
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment