超解像

1 files changed, 3 insertions, 4 deletions

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,