Merge pull request #3 from nakosung/patch-2

Filter out non .wav files from scanned file list
musyoku · Feb 17, 2017 · 39ce4f7 · 39ce4f7
2 parents 6098ea3 + acef21d
commit 39ce4f7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/train_audio/train.py b/train_audio/train.py
@@ -109,8 +109,10 @@ def main():
 	files = []
 	fs = os.listdir(args.wav_dir)
 	for fn in fs:
-		print "loading", fn
-		files.append(fn)
+		# filter out non-wav files
+		if fn.endswith('.wav'):
+			print "loading", fn
+			files.append(fn)
 
 	# compute receptive field width
 	num_layers = len(params.residual_conv_channels)