bridge.training.diffusion_gemma_step

Training step entry point for DiffusionGemma block-diffusion SFT.

Module Contents

Data

__all__

API

bridge.training.diffusion_gemma_step.__all__

[‘corrupt_canvas’, ‘diffusion_loss_function’, ‘forward_step’, ‘get_batch_from_iterator’]