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

Method on_train_batch_end

ldm/models/diffusion/ddpm.py:472–474  ·  view source on GitHub ↗
(self, *args, **kwargs)

Source from the content-addressed store, hash-verified

470 self.log_dict(loss_dict_ema, prog_bar=False, logger=True, on_step=False, on_epoch=True)
471
472 def on_train_batch_end(self, *args, **kwargs):
473 if self.use_ema:
474 self.model_ema(self.model)
475
476 def _get_rows_from_list(self, samples):
477 n_imgs_per_row = len(samples)

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected