Python -Получение ошибки AssertError при запуске обучающего модуля

Question

Я работаю над алгоритмом разделения динамиков и продолжаю получать следующую ошибку при запуске обучающего модуля:

  File "<ipython-input-9-3d6707b20beb>", line 1, in <module>
    runfile('C:/Users/User/Neural-mask-estimation/train.py', wdir='C:/Users/User/Neural-mask-estimation')

  File "C:\Users\User\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 827, in runfile
    execfile(filename, namespace)

  File "C:\Users\User\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 110, in execfile
    exec(compile(f.read(), filename, 'exec'), namespace)

  File "C:/Users/User/Neural-mask-estimation/train.py", line 87, in <module>
    is_adapt=False)

  File "C:\Users\User\Neural-mask-estimation\maskestimator\model.py", line 74, in get_model
    optimizer=optimizers.RMSprop(lr=self.lr, decay=1e-6, epsilon=1e-06, clipnorm=1.0))

  File "C:\Users\User\Anaconda3\lib\site-packages\tensorflow_core\python\training\tracking\base.py", line 457, in _method_wrapper
    result = method(self, *args, **kwargs)

  File "C:\Users\User\Anaconda3\lib\site-packages\tensorflow_core\python\keras\engine\training.py", line 446, in compile
    self._compile_weights_loss_and_weighted_metrics()

  File "C:\Users\User\Anaconda3\lib\site-packages\tensorflow_core\python\training\tracking\base.py", line 457, in _method_wrapper
    result = method(self, *args, **kwargs)

  File "C:\Users\User\Anaconda3\lib\site-packages\tensorflow_core\python\keras\engine\training.py", line 1574, in _compile_weights_loss_and_weighted_metrics
    self._prepare_sample_weights(sample_weights)

  File "C:\Users\User\Anaconda3\lib\site-packages\tensorflow_core\python\keras\engine\training.py", line 1831, in _prepare_sample_weights
    endpoint.populate_sample_weight(weight, endpoint.sample_weight_mode)

  File "C:\Users\User\Anaconda3\lib\site-packages\tensorflow_core\python\keras\engine\training.py", line 3119, in populate_sample_weight
    assert sample_weight_mode in ['temporal', 'samplewise']

AssertionError

Для справки, я запускаю файл train.py из https://github.com/AkojimaSLP/Neural-mask-estimation.git