da3a79468e
Replace the with-replacement per-batch index draw in the SDXL and DiT LoRA trainers with a shared PermutationBatchSampler that visits every image once per cycle before repeating, so short runs cover the whole dataset. The sampler reshuffles from the run's rng so the index stream stays seed-deterministic. Guard the diffusion run detail route against a valid-JSON non-object record, which previously raised TypeError and returned a 500; it now 404s like the list path's shape check. Add regression tests for both. |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| diffusion_dit_trainer.py | ||
| diffusion_lora_trainer.py | ||
| diffusion_train_common.py | ||
| diffusion_training_service.py | ||
| resume.py | ||
| s3_dataset.py | ||
| trainer.py | ||
| training.py | ||
| worker.py | ||