cc783998481f2882b3a2ec70d2f59d3cebc7be2b,official/nlp/modeling/models/bert_classifier.py,BertClassifier,checkpoint_items,#BertClassifier#,99

Before Change

  @property
  def checkpoint_items(self):
    return dict(encoder=self._network)

  def get_config(self):
    return self._config

After Change

  def checkpoint_items(self):
    items = dict(encoder=self._network)
    if hasattr(self.classifier, "checkpoint_items"):
      for key, item in self.classifier.checkpoint_items.items():
        items[".".join([self.classifier.name, key])] = item
    return items

  def get_config(self):
    return self._config

Instances