From 2be3f03adc5695f82c6ab86da780108f786ed014 Mon Sep 17 00:00:00 2001 From: Hiroshiba Kazuyuki Date: Sun, 14 Jan 2018 07:40:07 +0900 Subject: 超解像 MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- train.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) (limited to 'train.py') diff --git a/train.py b/train.py index 346ec4b..3e8cced 100644 --- a/train.py +++ b/train.py @@ -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, -- cgit v1.2.3-70-g09d2