Esempio n. 1
0
    def _cleanup(self):
        self._close_writers()

        if self.metrics_writer:
            self.metrics_writer.close()

        training_has_ended(self.out_dir)
Esempio n. 2
0
    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)
Esempio n. 3
0
    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)