add er+fn reverb, checkpoints

Author: erl-j

air_dataset.sh

@@ -1,9 +1,9 @@
-# #python scripts/prepare_dataset_2.py --input_audio_pattern=Saxophone --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
-# python scripts/prepare_dataset_2.py --input_audio_pattern=Bassoon  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
-# python scripts/prepare_dataset_2.py --input_audio_pattern=Clarinet  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
-# python scripts/prepare_dataset_2.py --input_audio_pattern=Flute  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
-# python scripts/prepare_dataset_2.py --input_audio_pattern=Horn  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
-# python scripts/prepare_dataset_2.py --input_audio_pattern=Oboe  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
-# python scripts/prepare_dataset_2.py --input_audio_pattern=Trombone  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
-# python scripts/prepare_dataset_2.py --input_audio_pattern=Trumpet  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
-# python scripts/prepare_dataset_2.py --input_audio_pattern=Tuba  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
+python scripts/prepare_dataset_2.py --input_audio_pattern=Saxophone --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
+python scripts/prepare_dataset_2.py --input_audio_pattern=Bassoon  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
+python scripts/prepare_dataset_2.py --input_audio_pattern=Clarinet  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
+python scripts/prepare_dataset_2.py --input_audio_pattern=Flute  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
+python scripts/prepare_dataset_2.py --input_audio_pattern=Horn  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
+python scripts/prepare_dataset_2.py --input_audio_pattern=Oboe  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
+python scripts/prepare_dataset_2.py --input_audio_pattern=Trombone  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
+python scripts/prepare_dataset_2.py --input_audio_pattern=Trumpet  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/
+python scripts/prepare_dataset_2.py --input_audio_pattern=Tuba  --sample_rate=48000 --output_tfrecord_path=datasets/AIR/tfr48k/

checkpoints/48k_bidir_z512_conv_family_Saxophone_f0c.data-00000-of-00001

@@ -27,6 +27,7 @@ def get_dataset(self, shuffle=False):
             instrument_dataset_provider = ddsp.training.data.TFRecordProvider(
                 f, self.example_secs, self.sample_rate, self.frame_rate)
             instrument_dataset = instrument_dataset_provider.get_dataset()
+            instrument_dataset=instrument_dataset.filter(lambda x: tf.math.count_nonzero(x["audio"])>0)
             instrument_dataset = instrument_dataset.map(
                 lambda x: {**x, "instrument": f, "instrument_idx": fi})
             if multi_dataset == None: