Unable to save keras model with multi worker distribution strategy