16 commits

Author	SHA1	Message	Date
Shen Chenhui	d71a06756b	Hotfix/vae sampler (#106) ... * format * format * fix eval loss * format * use default seed * format * change back ckpt_every to 1k --------- Co-authored-by: Shen-Chenhui <shen_chenhui@u.nus.edu>	2024-05-21 10:36:26 +08:00
Zheng Zangwei (Alex Zheng)	9cf394884c	Dev/load batch (#105) ... * add train_load_batch * update train_load_batch --------- Co-authored-by: pxy <pexure@gmail.com>	2024-05-20 17:40:05 +08:00
zhengzangw	5b9e753039	[feat] unify dataloader api	2024-05-20 08:40:45 +00:00
zhengzangw	5aad90acaf	[feat] extract feature	2024-05-17 09:53:48 +00:00
zhengzangw	9fd4554f55	[fix] vae dtype	2024-05-16 08:50:24 +00:00
zhengzangw	2a55d672a7	[feat] update logging and bucket	2024-05-09 08:07:56 +00:00
Zangwei Zheng	3944e93065	update training config	2024-04-22 12:05:59 +08:00
Zangwei Zheng	dce2ef4a1c	[exp] new config	2024-04-19 15:21:00 +08:00
Zheng Zangwei (Alex Zheng)	069ea0d687	Feat/fast bucket (#54) ... * [wip] bucket * [bug] not parallel * update eval * update sample.sh * accelerate bucket build with pandarallel	2024-04-19 11:42:02 +08:00
Zheng Zangwei (Alex Zheng)	7612d22fc6	Exp/image mix (#20) ... * [exp] image mixed training * [exp] add batch info * [exp] launch * update num_step_per_epoch * [feat] verify image mix training	2024-04-02 13:35:09 +08:00
Frank Lee	baf91f835e	updated sampler workload and state dict (#18)	2024-04-01 21:10:51 +08:00
Frank Lee	b8b78c6733	Hotfix/sampler (#16) ... * fixed variable video sampler * polish * polish	2024-04-01 20:37:41 +08:00
Frank Lee	a9e8220c7c	fixed image size in sampler (#15) ... * fixed image size in sampler * polish	2024-04-01 11:43:56 +08:00
Frank Lee	46f9af34fc	fixed multiprocessing dataloader (#14)	2024-03-31 20:59:22 +08:00
Zheng Zangwei (Alex Zheng)	f9f539f07e	format and some fix (#8)	2024-03-30 13:34:19 +08:00
Hongxin Liu	5e82d1493b	[feature] refactor sampler (#4) ... * [feature] refactor sampler * [feature] support sampler resuming * [feature] support sampler resuming	2024-03-29 23:32:12 +08:00