机器学习PAI中modelClass的 proto不变，这样可以吗？

机器学习PAI中modelClass的 proto不变
easy_rec_model的proto增加
modelClass newModel = xxx;

模型文件：
class newModel(BaseModel):

def init(self,
model_config,
feature_configs,
features,
labels=None,
is_training=False):
super(newModel, self).init(model_config, feature_configs, features,
labels, is_training)
self._losses = self._model_config.losses
assert self._model_config.WhichOneof(
'model'
) == 'newModel', 'invalid model config: %s' % self._model_config.WhichOneof(
'model')
self._model_config = self._model_config.newModel
assert isinstance(self._model_config, modelClassConfig)

模型配置：
model_config {
model_class: "newModel"
newModel {
}
这样可以吗？