def validation_epoch_end(self, output_data_batch): """Finishes a validation epoch.""" # Reduce depth metrics metrics_data = all_reduce_metrics(output_data_batch, self.validation_dataset, self.metrics_name) # Create depth dictionary metrics_dict = create_dict(metrics_data, self.metrics_keys, self.metrics_modes, self.config.datasets.validation) # Print stuff self.print_metrics(metrics_data, self.config.datasets.validation) # Log to wandb if self.logger: self.logger.log_metrics({ **metrics_dict, 'global_step': self.current_epoch + 1, }) return {**metrics_dict}
def test_epoch_end(self, output_data_batch, filename): """Finishes a test epoch.""" # Reduce depth metrics metrics_data = all_reduce_metrics(output_data_batch, self.test_dataset, self.metrics_name) # Create depth dictionary metrics_dict = create_dict(metrics_data, self.metrics_keys, self.metrics_modes, self.config.datasets.test) np.save(filename, metrics_dict) # Print stuff self.print_metrics(metrics_data, self.config.datasets.test) return {**metrics_dict}