def save_generation(self, weights): # Weights is a list of lists of numpy.ndarrays for idx, weight_array in enumerate(weights): numpy.savez(self.get_save_file(idx), *weight_array)