-
Notifications
You must be signed in to change notification settings - Fork 471
Closed
Description
I ran train.py and this is what I got
Using TensorFlow backend.
2018-02-27 12:22:38.975784: I C:\tf_jenkins\home\workspace\rel-win\M\windows\PY\36\tensorflow
\core\platform\cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow
binary was not compiled to use: AVX AVX2
Loading LSTM model.
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
lstm_1 (LSTM) (None, 2048) 33562624
_________________________________________________________________
dense_1 (Dense) (None, 512) 1049088
_________________________________________________________________
dropout_1 (Dropout) (None, 512) 0
_________________________________________________________________
dense_2 (Dense) (None, 14) 7182
=================================================================
Total params: 34,618,894
Trainable params: 34,618,894
Non-trainable params: 0
_________________________________________________________________
None
Creating train generator with 850 samples.
Epoch 1/1000
Exception in thread Thread-2:
Traceback (most recent call last):
File "C:\Users\Munic\AppData\Local\Programs\Python\Python36\lib\threading.py", line 916, in
_bootstrap_inner
self.run()
File "C:\Users\Munic\AppData\Local\Programs\Python\Python36\lib\threading.py", line 864, in
run
self._target(*self._args, **self._kwargs)
File "C:\Users\Munic\AppData\Local\Programs\Python\Python36\lib\site-packages\keras\utils\d
ata_utils.py", line 579, in data_generator_task
generator_output = next(self._generator)
File "C:\Users\Munic\Documents\Popping-Dance-Style-Classification\data.py", line 25, in __n
ext__
return next(self.iterator)
File "C:\Users\Munic\Documents\Popping-Dance-Style-Classification\data.py", line 191, in fr
ame_generator
y.append(self.get_class_one_hot(sample[1]))
File "C:\Users\Munic\Documents\Popping-Dance-Style-Classification\data.py", line 103, in ge
t_class_one_hot
assert len(label_hot) == len(self.classes)
AssertionError
Traceback (most recent call last):
File "C:\Users\Munic\Documents\Popping-Dance-Style-Classification\train.py", line 110, in <
module>
main()
File "C:\Users\Munic\Documents\Popping-Dance-Style-Classification\train.py", line 107, in m
ain
load_to_memory=load_to_memory, batch_size=batch_size, nb_epoch=nb_epoch)
File "C:\Users\Munic\Documents\Popping-Dance-Style-Classification\train.py", line 81, in tr
ain
workers=4)
File "C:\Users\Munic\AppData\Local\Programs\Python\Python36\lib\site-packages\keras\legacy\
interfaces.py", line 87, in wrapper
return func(*args, **kwargs)
File "C:\Users\Munic\AppData\Local\Programs\Python\Python36\lib\site-packages\keras\models.
py", line 1156, in fit_generator
initial_epoch=initial_epoch)
File "C:\Users\Munic\AppData\Local\Programs\Python\Python36\lib\site-packages\keras\legacy\
interfaces.py", line 87, in wrapper
return func(*args, **kwargs)
File "C:\Users\Munic\AppData\Local\Programs\Python\Python36\lib\site-packages\keras\engine\
training.py", line 2046, in fit_generator
generator_output = next(output_generator)
StopIteration
What could have went wrong?
Metadata
Metadata
Assignees
Labels
No labels