def _state_dict_to_master_params()

in improved_diffusion/train_util.py [0:0]

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

    def _state_dict_to_master_params(self, state_dict):
        params = [state_dict[name] for name, _ in self.model.named_parameters()]
        if self.use_fp16:
            return make_master_params(params)
        else:
            return params