4 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
gameloader	50b1f76263	feat(models/denoiser): Add Mamba2 model implementation ... Implement a minimal, single-file implementation of the Mamba-2 model in PyTorch. Based on the paper "Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Duality".	2026-01-20 14:44:03 +08:00
gameloader	b9c3720eba	feat(callbacks): add Weights & Biases image logging to SaveImagesHook\n\n- Introduce optional wandb_max_save_num parameter in __init__\n- Add helper methods _get_wandb_logger and _log_wandb_samples\n- Log sampled images as wandb.Image artifacts during validation epoch end (global zero only)\n- Handle proper dtype conversion and clipping for wandb compatibility\n- Respect both general max_save_num and wandb-specific limit\n\nThis enables visualization of validation samples directly in the W&B dashboard without manual intervention.	2026-01-19 16:03:54 +08:00
game-loader	6bf32b08fd	feat(dataset): add HuggingFace image dataset	2026-01-18 17:10:01 +08:00
game-loader	cf45afe325	feat(data): add support for Hugging Face datasets	2026-01-16 14:25:32 +08:00