def get_network_description(network):
        """Get the string and total parameters of the network"""
        # pdb.set_trace()
        # network = network.module
        return str(network), sum(map(lambda x: x.numel(), network.parameters()))