def num_trainable_parameters()

in utils/helpers.py [0:0]

• 2 lines of code
• 3 McCabe index (conditional complexity)

    def num_trainable_parameters(self):
        """
        :return: number of trainable parameters in the model
        """
        return sum(p.numel() for p in self.parameters() if p.requires_grad)