diff options
Diffstat (limited to 'train.py')
| -rw-r--r-- | train.py | 7 |
1 files changed, 3 insertions, 4 deletions
@@ -10,10 +10,10 @@ from chainer.iterators import MultiprocessIterator from chainer.training import extensions from chainerui.utils import save_args -from become_yukarin.config import create_from_json +from become_yukarin.config.config import create_from_json from become_yukarin.dataset import create as create_dataset -from become_yukarin.model import create -from become_yukarin.updater import Updater +from become_yukarin.model.model import create +from become_yukarin.updater.updater import Updater parser = argparse.ArgumentParser() parser.add_argument('config_json_path', type=Path) @@ -54,7 +54,6 @@ opts = {key: create_optimizer(model) for key, model in models.items()} converter = partial(convert.concat_examples, padding=0) updater = Updater( loss_config=config.loss, - model_config=config.model, predictor=predictor, aligner=aligner, discriminator=discriminator, |