MCPcopy Create free account
hub / github.com/dome272/Diffusion-Models-pytorch / reset_parameters

Method reset_parameters

modules.py:31–32  ·  view source on GitHub ↗
(self, ema_model, model)

Source from the content-addressed store, hash-verified

29 self.step += 1
30
31 def reset_parameters(self, ema_model, model):
32 ema_model.load_state_dict(model.state_dict())
33
34
35class SelfAttention(nn.Module):

Callers 1

step_emaMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected