yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-31 14:58:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: add probe in train

Browse Source
This commit is contained in:
Félix Voituret
2019-11-20 15:40:25 +01:00
parent 151c2a7b93
commit fbe5f290ef
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
spleeter/commands/train.py
View File

@@ -16,6 +16,7 @@ import tensorflow as tf
from ..audio.adapter import get_audio_adapter from ..audio.adapter import get_audio_adapter
from ..dataset import get_training_dataset, get_validation_dataset from ..dataset import get_training_dataset, get_validation_dataset
from ..model import model_fn from ..model import model_fn
from ..model.provider import ModelProvider
from ..utils.logging import get_logger from ..utils.logging import get_logger
__email__ = 'research@deezer.com' __email__ = 'research@deezer.com'
@@ -95,4 +96,5 @@ def entrypoint(arguments, params):
estimator, estimator,
train_spec, train_spec,
evaluation_spec) evaluation_spec)
ModelProvider.writeProbe(params['model_dir'])
get_logger().info('Model training done') get_logger().info('Model training done')

6
spleeter/model/provider/__init__.py
View File

@@ -38,12 +38,14 @@ class ModelProvider(ABC):
""" """
pass pass
def writeProbe(self, directory): @staticmethod
def writeProbe(directory):
""" Write a model probe file into the given directory. """ Write a model probe file into the given directory.
:param directory: Directory to write probe into. :param directory: Directory to write probe into.
""" """
with open(join(directory, self.MODEL_PROBE_PATH), 'w') as stream: probe = join(directory, ModelProvider.MODEL_PROBE_PATH)
with open(probe, 'w') as stream:
stream.write('OK') stream.write('OK')
def get(self, model_directory): def get(self, model_directory):