Training leaf_custom issue!

[ahmadrv]

Hi there,
When i try to run this:
python3 -m example.main --gin_config=example/configs/leaf_custom.gin

returns:

  TypeError: in user code:

  /usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/engine/training.py:855 train_function  *
      return step_function(self, iterator)
  /content/leaf-audio/leaf_audio/models.py:54 call  *
      output = self._frontend(output, training=training)  # pylint: disable=not-callable
  /content/leaf-audio/leaf_audio/frontend.py:199 call  *
      outputs = self._spec_augment_fn(outputs)
  /content/leaf-audio/leaf_audio/postprocessing.py:192 call  *
      outputs = self.spec_augment_layer._AugmentationNetwork(  # pylint: disable=protected-access

  TypeError: tf___AugmentationNetwork() got an unexpected keyword argument 'series_length'

In call to configurable 'train' (<function train at 0x7f5d19819050>)

Would you show me the problem?

Additionally, running on Colab.

[lienz]

Hi, the issue was due to lingvo being upgraded to 0.9.0.
This should be fixed by our latest commit, please let us know if you still face issues.

[ahmadrv]

Hi, the issue was due to lingvo being upgraded to 0.9.0. This should be fixed by our latest commit, please let us know if you still face issues.

Thanks a lot. That is all you mentioned.