def after_run()

in benchmarks/horovod-resnet/train_imagenet_resnet_hvd.py [0:0]

• 14 lines of code
• 3 McCabe index (conditional complexity)

    def after_run(self, run_context, run_values):
        self.elapsed_secs += time.time() - self.t0
        self.count += 1
        global_step, loss, total_loss, lr = run_values.results
        if global_step == 1 or global_step % self.display_every == 0:
            dt = self.elapsed_secs / self.count
            img_per_sec = self.global_batch_size / dt
            epoch = global_step * self.global_batch_size / self.num_records
            self.logger.info(
                "%6i %5.1f %7.1f %6.3f %6.3f %7.5f"
                % (global_step, epoch, img_per_sec, loss, total_loss, lr)
            )
            self.elapsed_secs = 0.0
            self.count = 0