def step_frozen()

in agents/diffusion.py [0:0]

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

    def step_frozen(self):
        for param, target_param in zip(self.model.parameters(), self.model_frozen.parameters()):
            target_param.data.copy_(param.data)