def _cleanup(self):
self._close_writers()
if self.metrics_writer:
self.metrics_writer.close()
training_has_ended(self.out_dir)
if self.first_process is True:
remove_claim_file(self.out_dir)
def _cleanup(self):
self._close_writers()
if self.metrics_writer:
self.metrics_writer.close()
self.log_outstanding_timeline_metrics()
self.timeline_writer.close()
# close the Horovod file reader thread if it has been enabled
if self.hvd_reader and self.hvd_reader.enabled:
self.hvd_reader.close()
training_has_ended(self.out_dir)
if self.first_process is True:
remove_claim_file(self.out_dir)