def set_estimator(self, estimator):
    super(StepCounter, self).set_estimator(estimator)
    if (self._summary_writer is None):
        self._summary_writer = SummaryWriterCache.get(estimator.model_dir)
def set_estimator(self, estimator):
    super(StepCounter, self).set_estimator(estimator)
    if (self._summary_writer is None):
        self._summary_writer = SummaryWriterCache.get(estimator.model_dir)