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
wangshuai6
ae7df43ecc
torch.compile
2025-07-03 20:23:59 +08:00
wang shuai
2c16b8f423
Update improved_dit.py
2025-07-03 20:21:04 +08:00
wangshuai6
8038c16bee
disperse loss
2025-06-14 12:14:01 +08:00
wangshuai6
598f7b40a2
disperse loss
2025-06-13 10:41:31 +08:00
wangshuai6
3093a65151
disperse loss
2025-06-12 22:18:15 +08:00
wangshuai6
99d92c94e7
fix bugs(admas timedeltas)
2025-05-20 12:30:40 +08:00
wangshuai6
3693640ca3
fix bugs(admas timedeltas)
2025-05-20 12:27:34 +08:00
wangshuai6
4d4b3262c3
fix bugs(admas timedeltas)
2025-05-20 12:25:50 +08:00
wangshuai6
9010659f72
fix bugs(update src)
2025-04-14 08:45:47 +08:00
wangshuai6
ce93cc61a8
fix bugs(rm encoder.py)
2025-04-13 09:14:35 +08:00
wangshuai6
687d650175
app demo
2025-04-11 19:14:51 +08:00
wangshuai6
485818abf6
remove hardcode dino encoder logic
2025-04-11 11:07:59 +08:00
wangshuai6
5c3867ffdc
dynamic programming
2025-04-10 10:02:41 +08:00
wangshuai6
d805c7cb71
delete callbacks
2025-04-09 11:15:09 +08:00
wangshuai6
4f60297dfe
decoupled dit clean code
2025-04-09 11:14:14 +08:00
wangshuai6
d1b6da1f0a
decoupled dit src code
2025-04-09 11:07:29 +08:00
wangshuai6
06499f1caa
submit code
2025-04-09 11:01:16 +08:00
