Error executing with --stft

[cperezs]

Hi,
I'm trying to run the sample experiments with MusicNet, but replacing the --fourier param with --stft:
train.py deep_model --in-memory --model=deep_convnet --local-data data/musicnet_11khz.npz --fast-load --stft

No matter which model I use, when using --stft I always get an error like this:

.. building model
.. using deep convnet
Traceback (most recent call last):
  File "./train.py", line 142, in <module>
    main(**parser.parse_args().__dict__)
  File "./train.py", line 100, in main
    model = get_model(model, dataset.feature_dim)
  File "./train.py", line 77, in get_model
    channels=feature_dim[1])
  File "/home/cperez/code/audio-transcription-complex-networks/musicnet/musicnet/models/__init__.py", line 95, in get_deep_convnet
    kernel_initializer='glorot_normal'))
  File "/home/cperez/miniconda3/envs/complex/lib/python2.7/site-packages/keras/models.py", line 455, in add
    output_tensor = layer(self.outputs[0])
  File "/home/cperez/miniconda3/envs/complex/lib/python2.7/site-packages/keras/engine/topology.py", line 528, in __call__
    self.build(input_shapes[0])
  File "/home/cperez/miniconda3/envs/complex/lib/python2.7/site-packages/keras/layers/core.py", line 827, in build
    constraint=self.kernel_constraint)
  File "/home/cperez/miniconda3/envs/complex/lib/python2.7/site-packages/keras/engine/topology.py", line 364, in add_weight
    weight = K.variable(initializer(shape), dtype=K.floatx(), name=name)
  File "/home/cperez/miniconda3/envs/complex/lib/python2.7/site-packages/keras/initializers.py", line 201, in __call__
    dtype=dtype, seed=self.seed)
  File "/home/cperez/miniconda3/envs/complex/lib/python2.7/site-packages/keras/backend/theano_backend.py", line 1987, in truncated_normal
    normal_tensor = rng.normal(size=shape, avg=mean, std=stddev, dtype=dtype)
  File "/home/cperez/miniconda3/envs/complex/lib/python2.7/site-packages/theano/sandbox/rng_mrg.py", line 1574, in normal
    nstreams=nstreams)
  File "/home/cperez/miniconda3/envs/complex/lib/python2.7/site-packages/theano/sandbox/rng_mrg.py", line 1344, in uniform
    size)
ValueError: ('The specified size contains a dimension with value <= 0', (-1572864,))

Is there something wrong in the way I'm using the script? Notice I'm using the --fast-load param, I have not tried with the full dataset yet.
Thanks!

[dmitriy-serdyuk]

Yes, there was a problem with --fast-load, try to load the entire dataset into memory, it is about 30G.