CyberZHG/keras-radam

Unknown optimizer: RAdam

stvogel opened this issue · 3 comments

stvogel commented

I'm currently using Keras/TF 1.12.
Model is trained with optimizer RAdam and saved.
When I load the model again with keras.models.load_model(x)
I get:

  File "src/models/eval_model.py", line 105, in <module>
    model = keras.models.load_model(args.model_filepath)
  File "keras/engine/saving.py", line 419, in load_model
    model = _deserialize_model(f, custom_objects, compile)
  File "keras/engine/saving.py", line 299, in _deserialize_model
    custom_objects=custom_objects)
  File "keras/optimizers.py", line 768, in deserialize
    printable_module_name='optimizer')
  File "keras/utils/generic_utils.py", line 138, in deserialize_keras_object
    ': ' + class_name)
ValueError: Unknown optimizer: RAdam

I did an import: from keras_radam import RAdam
What am I doing wrong?

CyberZHG commented

keras-radam/tests/test_optimizers.py

Line 40 in 78dcb11

model = keras.models.load_model(model_path, custom_objects={'RAdam': RAdam})

  • 👍15
stvogel commented

Dooh. Somehow I thought, this might be a dummy question.
Thanks a lot for your help!

FrancisDacian commented

keras-radam/tests/test_optimizers.py

Line 40 in 78dcb11

model = keras.models.load_model(model_path, custom_objects={'RAdam': RAdam})

Thanks.