search

salu133445 / bmusegan

Code for “Convolutional Generative Adversarial Networks with Binary Neurons for Polyphonic Music Generation”
https://salu133445.github.io/bmusegan/
MIT License
58 stars 13 forks source link

Is it need two stage train #1

Closed ericosmic closed 5 years ago

ericosmic commented 5 years ago

If musegan/bmusegan is need two stages train and each stage need change some config parameters (like stage_phase)?

salu133445 commented 5 years ago

You need to modify the following values of dictionary SETUP in config.py.

Key Value (Stage I) Value (Stage II)
'two_stage_training' True True
'training_phase' 'pretrain' 'train'
'first_stage_dir' None 'path/to/pretrained/model'

(Usually, SETUP['first_stage_dir'] can be automatically determined. Just set it to None and see if it works. If not, then specify it manually.)