diff options
Diffstat (limited to 'generate.py')
| -rw-r--r-- | generate.py | 20 |
1 files changed, 17 insertions, 3 deletions
diff --git a/generate.py b/generate.py index c4059ec..f51b586 100644 --- a/generate.py +++ b/generate.py @@ -52,7 +52,10 @@ default_params = { 'loss_smoothing': 0.99, 'cuda': True, 'comet_key': None, - 'primer': '' + 'primer': '', + 'primer_a': '', + 'primer_b': '', + 'recursive': True, } tag_params = [ @@ -224,7 +227,8 @@ def main(exp, frame_sizes, dataset, **params): trainer.register_plugin(GeneratorPlugin( os.path.join(results_path, 'samples'), params['n_samples'], params['sample_length'], params['sample_rate'], - params['primer'] + params['primer'], params['primer_a'], params['primer_b'], + params['recursive'], )) """ trainer.register_plugin( @@ -362,7 +366,17 @@ if __name__ == '__main__': '--comet_key', help='comet.ml API key' ) parser.add_argument( - '--primer', help='prime the generator...' + '--primer', help='prime the generator with a function. noise, sin, some others..' + ) + parser.add_argument( + '--primer_a', help='optional first parameter for primer' + ) + parser.add_argument( + '--primer_b', help='optional second parameter for primer' + ) + parser.add_argument( + '--recursive', type=parse_bool, default=True, + help='generate output recursively' ) parser.set_defaults(**default_params) |