shamangary / FSA-Net

[CVPR19] FSA-Net: Learning Fine-Grained Structure Aggregation for Head Pose Estimation from a Single Image
Apache License 2.0
607 stars 153 forks source link

AttributeError: 'ModelCheckpoint' object has no attribute '_implements_train_batch_hooks' #77

Closed Algabri closed 1 year ago

Algabri commented 1 year ago

Hi

I am trying to run the training code but am facing a lot of errors.

File "/home/redhwan/2/HPE/FSA-Net/training_and_testing/FSANET_train.py", line 298, in <module>
    main()
  File "/home/redhwan/2/HPE/FSA-Net/training_and_testing/FSANET_train.py", line 286, in main
    hist = model.fit_generator(generator=data_generator_pose(X=x_train, Y=y_train, batch_size=batch_size),
  File "/home/redhwan/.local/lib/python3.8/site-packages/tensorflow/python/util/deprecation.py", line 324, in new_func
    return func(*args, **kwargs)
  File "/home/redhwan/.local/lib/python3.8/site-packages/tensorflow/python/keras/engine/training.py", line 1815, in fit_generator
    return self.fit(
  File "/home/redhwan/.local/lib/python3.8/site-packages/tensorflow/python/keras/engine/training.py", line 108, in _method_wrapper
    return method(self, *args, **kwargs)
  File "/home/redhwan/.local/lib/python3.8/site-packages/tensorflow/python/keras/engine/training.py", line 1067, in fit
    callbacks = callbacks_module.CallbackList(
  File "/home/redhwan/.local/lib/python3.8/site-packages/tensorflow/python/keras/callbacks.py", line 234, in __init__
    self._should_call_train_batch_hooks = any(
  File "/home/redhwan/.local/lib/python3.8/site-packages/tensorflow/python/keras/callbacks.py", line 235, in <genexpr>
    cb._implements_train_batch_hooks() for cb in self.callbacks)
AttributeError: 'ModelCheckpoint' object has no attribute '_implements_train_batch_hooks'

When I replacedfrom keras.callbacks import ModelCheckpoint by from tensorflow.keras.callbacks import ModelCheckpoint

I got this error:

Traceback (most recent call last):
  File "/home/redhwan/2/HPE/FSA-Net/training_and_testing/FSANET_train.py", line 298, in <module>
    main()
  File "/home/redhwan/2/HPE/FSA-Net/training_and_testing/FSANET_train.py", line 286, in main
    hist = model.fit_generator(generator=data_generator_pose(X=x_train, Y=y_train, batch_size=batch_size),
  File "/home/redhwan/.local/lib/python3.8/site-packages/tensorflow/python/util/deprecation.py", line 324, in new_func
    return func(*args, **kwargs)
  File "/home/redhwan/.local/lib/python3.8/site-packages/tensorflow/python/keras/engine/training.py", line 1815, in fit_generator
    return self.fit(
  File "/home/redhwan/.local/lib/python3.8/site-packages/tensorflow/python/keras/engine/training.py", line 108, in _method_wrapper
    return method(self, *args, **kwargs)
  File "/home/redhwan/.local/lib/python3.8/site-packages/tensorflow/python/keras/engine/training.py", line 1067, in fit
    callbacks = callbacks_module.CallbackList(
  File "/home/redhwan/.local/lib/python3.8/site-packages/tensorflow/python/keras/callbacks.py", line 236, in __init__
    self._should_call_test_batch_hooks = any(
  File "/home/redhwan/.local/lib/python3.8/site-packages/tensorflow/python/keras/callbacks.py", line 237, in <genexpr>
    cb._implements_test_batch_hooks() for cb in self.callbacks)
AttributeError: 'DecayLearningRate' object has no attribute '_implements_test_batch_hooks'

Could you run your code on your PC right now? PLs.

Keras ==2.2.4 as yours.