def default_hparams():
return tf_lib.HParams(batch_size=128,
rnn_layer_sizes=[128, 128],
dropout_keep_prob=0.5,
skip_first_n_losses=0,
clip_norm=5,
initial_learning_rate=0.01,
decay_steps=1000,
decay_rate=0.95)
def setUp(self):
self.encoder_decoder = melodies_lib.OneHotEncoderDecoder(0, 12, 0)
self.hparams = tf_lib.HParams(batch_size=128,
rnn_layer_sizes=[128, 128],
dropout_keep_prob=0.5,
skip_first_n_losses=0,
clip_norm=5,
initial_learning_rate=0.01,
decay_steps=1000,
decay_rate=0.85)