Another workaround is to use RectifiedAdam optimizer from tensorflow-addon, if SGD does not work as well.
Topic:
Machine Learning & AI
SubTopic:
General
Tags: