MCPcopy Create free account
hub / github.com/VisionXLab/OF-Diff / on_train_batch_end

Method on_train_batch_end

ldm/models/autoencoder.py:85–87  ·  view source on GitHub ↗
(self, *args, **kwargs)

Source from the content-addressed store, hash-verified

83 print(f"{context}: Restored training weights")
84
85 def on_train_batch_end(self, *args, **kwargs):
86 if self.use_ema:
87 self.model_ema(self)
88
89
90 def encode(self, x):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected