def num_trainable_parameters()

in src/utils.py [0:0]

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

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