Skip to content

Commit 238a0bc

Browse files
authored
Merge pull request lisa-lab#201 from LeoMingo/patch-3
Update rnnrbm.py
2 parents 8bd8a5a + e82dadc commit 238a0bc

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

code/rnnrbm.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -288,7 +288,8 @@ def generate(self, filename, show=True):
288288

289289
def test_rnnrbm(batch_size=100, num_epochs=200):
290290
model = RnnRbm()
291-
re = os.path.join(os.path.split(os.path.dirname(__file__))[0],
291+
cwd = os.path.dirname(os.path.abspath(__file__))
292+
re = os.path.join(os.path.split(cwd)[0],
292293
'data', 'Nottingham', 'train', '*.mid')
293294
model.train(glob.glob(re),
294295
batch_size=batch_size, num_epochs=num_epochs)

0 commit comments

Comments
 (0)