Skip to content

Commit

Permalink
Hyperparameter changes
Browse files Browse the repository at this point in the history
  • Loading branch information
calclavia committed May 3, 2017
1 parent c2524ef commit 642cc40
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions constants.py
Original file line number Diff line number Diff line change
@@ -28,14 +28,14 @@

# Training parameters
BATCH_SIZE = 32
SEQ_LEN = NOTES_PER_BAR
SEQ_LEN = 4 * NOTES_PER_BAR

# Hyper Parameters
OCTAVE_UNITS = 32
STYLE_UNITS = 32
BEAT_UNITS = 32
TIME_AXIS_UNITS = 256
NOTE_AXIS_UNITS = 128
TIME_AXIS_UNITS = 300
NOTE_AXIS_UNITS = 150

TIME_AXIS_LAYERS = 2
NOTE_AXIS_LAYERS = 2
2 changes: 1 addition & 1 deletion generate.py
Original file line number Diff line number Diff line change
@@ -7,7 +7,7 @@
from tqdm import tqdm
from midi_util import midi_encode

def generate(model, style=[0.25, 0.25, 0.25, 0.25], num_bars=16, default_temp=1):
def generate(model, style=[1, 0, 0, 0], num_bars=16, default_temp=1):
print('Generating')
notes_memory = deque([np.zeros((NUM_NOTES, 2)) for _ in range(SEQ_LEN)], maxlen=SEQ_LEN)
beat_memory = deque([np.zeros_like(compute_beat(0, NOTES_PER_BAR)) for _ in range(SEQ_LEN)], maxlen=SEQ_LEN)

0 comments on commit 642cc40

Please sign in to comment.