From 6a84007c044c1664eb998f1b4fd95c6af9878fe2 Mon Sep 17 00:00:00 2001 From: Hiroshiba Kazuyuki Date: Tue, 27 Feb 2018 21:52:11 +0900 Subject: pix2pixモデルのフィルタサイズを可変にした MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- become_yukarin/config/sr_config.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'become_yukarin/config/sr_config.py') diff --git a/become_yukarin/config/sr_config.py b/become_yukarin/config/sr_config.py index 4f980a2..75cf6ff 100644 --- a/become_yukarin/config/sr_config.py +++ b/become_yukarin/config/sr_config.py @@ -20,7 +20,10 @@ class SRDatasetConfig(NamedTuple): class SRModelConfig(NamedTuple): - pass + generator_base_channels: int + generator_extensive_layers: int + discriminator_base_channels: int + discriminator_extensive_layers: int class SRLossConfig(NamedTuple): @@ -85,6 +88,10 @@ def create_from_json(s: Union[str, Path]): num_test=d['dataset']['num_test'], ), model=SRModelConfig( + generator_base_channels=d['model']['generator_base_channels'], + generator_extensive_layers=d['model']['generator_extensive_layers'], + discriminator_base_channels=d['model']['discriminator_base_channels'], + discriminator_extensive_layers=d['model']['discriminator_extensive_layers'], ), loss=SRLossConfig( mse=d['loss']['mse'], -- cgit v1.2.3-70-g09d2