Closed soumen-ghosh closed 2 years ago
After using below code in the test function, the restoration worked - running prediction on test dataset:
model = tf.keras.models.load_model(modelFileName, custom_objects={'AdaBeliefOptimizer': AdaBeliefOptimizer})
Hi,
I've installed "adabelief-tf==0.2.0" through conda. And the tensorflow version I'm using is 2.8.0 (tensorflow==2.8.0). I'm facing below restoration issue while trying to try the model on test dataset - the checkpoint was created during training -
ValueError: Unknown optimizer: AdaBeliefOptimizer. Please ensure this object is passed to the
custom_objectsargument.
I'm using it below way:
I want to explore CosineDecayRestarts and AdaBeliefOptimizer together. Is there something wrong with my usage?