def get_virtual_batch()

in evaluations/th_evaluator.py [0:0]

• 18 lines of code
• 2 McCabe index (conditional complexity)

    def get_virtual_batch(self, data, num_samples, global_batch_size, resolution):
        preds, spatial_preds, clip_preds, batch = self.get_activations(
            data, num_samples, global_batch_size, pr_samples=10000
        )
        if self.is_root:
            fid_stats = self.get_statistics(preds, resolution)
            spatial_stats = self.get_statistics(spatial_preds, resolution)
            clip_stats = self.get_statistics(clip_preds, resolution)
            return batch, dict(
                mu=fid_stats.mu,
                sigma=fid_stats.sigma,
                mu_s=spatial_stats.mu,
                sigma_s=spatial_stats.sigma,
                mu_clip=clip_stats.mu,
                sigma_clip=clip_stats.sigma,
            )
        else:
            return None, dict()