Method param_count

bvr/training/train_bc.py:161–163

Total number of trainable parameters.

(self)

        return layers

    def param_count(self) -> int:
        """Total number of trainable parameters."""
        return sum(p.numel() for p in self.parameters())


def train(