Clean up of the NeuralNetworkDistributionStrategy.