def collect_parameters()

in projects/deep_video_compression/_utils.py [0:0]

• 12 lines of code
• 7 McCabe index (conditional complexity)

    def collect_parameters(self):
        model_parameters = {
            n: p
            for n, p in self.named_parameters()
            if not n.endswith(".quantiles") and p.requires_grad
        }
        quantile_parameters = {
            n: p
            for n, p in self.named_parameters()
            if n.endswith(".quantiles") and p.requires_grad
        }

        return model_parameters, quantile_parameters