diff options
| -rw-r--r-- | recipe/config.json | 44 | ||||
| -rw-r--r-- | recipe/config_sr.json | 31 | ||||
| -rw-r--r-- | recipe/recipe.json | 36 |
3 files changed, 111 insertions, 0 deletions
diff --git a/recipe/config.json b/recipe/config.json new file mode 100644 index 0000000..a559299 --- /dev/null +++ b/recipe/config.json @@ -0,0 +1,44 @@ +{ + "dataset": { + "features": [ + "f0", + "mfcc" + ], + "input_glob": "/hiho-pause/hiho-pause-atr-*.npy", + "input_global_noise": 0.01, + "input_local_noise": 0.01, + "input_mean_path": "/hiho-pause/mean.npy", + "input_var_path": "/hiho-pause/var.npy", + "num_test": 1, + "seed": 0, + "target_glob": "/yukari-pause/yukari-pause-atr-*.npy", + "target_global_noise": 0.01, + "target_local_noise": 0.01, + "target_mean_path": "/yukari-pause/mean.npy", + "target_var_path": "/yukari-pause/var.npy", + "train_crop_size": 512 + }, + "loss": { + "adversarial": 1, + "mse": 100 + }, + "model": { + "in_channels": 10, + "out_channels": 10, + "generator_base_channels": 64, + "generator_extensive_layers": 8, + "discriminator_base_channels": 32, + "discriminator_extensive_layers": 5, + "weak_discriminator": false + }, + "project": { + "name": "", + "tags": [] + }, + "train": { + "batchsize": 8, + "gpu": -1, + "log_iteration": 250, + "snapshot_iteration": 5000 + } +} diff --git a/recipe/config_sr.json b/recipe/config_sr.json new file mode 100644 index 0000000..3f290d9 --- /dev/null +++ b/recipe/config_sr.json @@ -0,0 +1,31 @@ +{ + "dataset": { + "input_glob": "/yukari-news-harvest/yukari-news-*.npy", + "num_test": 5, + "input_global_noise": 3, + "input_local_noise": 3, + "blur_size_factor": 0, + "seed": 0, + "train_crop_size": 512, + "generator_base_channels": 64, + "generator_extensive_layers": 8, + "discriminator_base_channels": 32, + "discriminator_extensive_layers": 5 + }, + "loss": { + "mse": 100, + "adversarial": 1 + }, + "model": { + }, + "project": { + "name": "", + "tags": [] + }, + "train": { + "batchsize": 8, + "gpu": -1, + "log_iteration": 250, + "snapshot_iteration": 5000 + } +}
\ No newline at end of file diff --git a/recipe/recipe.json b/recipe/recipe.json new file mode 100644 index 0000000..f0bc5d9 --- /dev/null +++ b/recipe/recipe.json @@ -0,0 +1,36 @@ +{ + "each": { + "model/generator_extensive_layers": [ + 8, + 0, + 8, + 0 + ], + "model/discriminator_extensive_layers": [ + 5, + 0, + 5, + 0 + ], + "model/weak_discriminator": [ + true, + true, + false, + false + ], + "train/gpu": [ + 0, + 1, + 2, + 3 + ], + "project/name": [ + "pp-weakD-el8", + "pp-weakD-el0", + "pp-el8", + "pp-el0" + ] + }, + "all": { + } +} |