Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Anthony Larcher
sidekit
Commits
383393c6
Commit
383393c6
authored
Apr 12, 2021
by
Anthony Larcher
Browse files
cleaning
parent
b17d1364
Changes
1
Hide whitespace changes
Inline
Side-by-side
nnet/xvector.py
View file @
383393c6
...
@@ -960,6 +960,7 @@ def get_network(model_opts, local_rank):
...
@@ -960,6 +960,7 @@ def get_network(model_opts, local_rank):
"""
"""
:param model_opts:
:param model_opts:
:param local_rank:
:return:
:return:
"""
"""
...
@@ -1011,7 +1012,9 @@ def get_network(model_opts, local_rank):
...
@@ -1011,7 +1012,9 @@ def get_network(model_opts, local_rank):
def
get_loaders
(
dataset_opts
,
training_opts
,
model_opts
):
def
get_loaders
(
dataset_opts
,
training_opts
,
model_opts
):
"""
"""
:param dataset_yaml:
:param dataset_opts:
:param training_opts:
:param model_opts:
:return:
:return:
"""
"""
...
@@ -1259,8 +1262,7 @@ def xtrain(dataset_description,
...
@@ -1259,8 +1262,7 @@ def xtrain(dataset_description,
monitor
.
logger
.
info
(
yaml
.
dump
(
training_opts
,
default_flow_style
=
False
))
monitor
.
logger
.
info
(
yaml
.
dump
(
training_opts
,
default_flow_style
=
False
))
# Initialize the model
# Initialize the model
model
=
get_network
(
model_opts
)
model
=
get_network
(
model_opts
,
local_rank
)
#speaker_number = model.speaker_number
embedding_size
=
model
.
embedding_size
embedding_size
=
model
.
embedding_size
# Set the device and manage parallel processing
# Set the device and manage parallel processing
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment