90 commits

Author	SHA1	Message	Date
zhengzangw	ce601066c5	[stage2] update config	2024-06-05 15:49:51 +00:00
Frank Lee	ea96cdb5e4	added warmup lr scheduler (#121)	2024-06-05 10:12:52 +08:00
Zheng Zangwei (Alex Zheng)	9370fbe02d	update timer (#115)	2024-05-29 14:46:14 +08:00
Frank Lee	2e1e26a2e4	added timer for benchmarking (#114)	2024-05-29 14:43:15 +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	a016325a8e	[feat] enable freezing y_embedder	2024-05-17 06:40:44 +00:00
zhengzangw	9fd4554f55	[fix] vae dtype	2024-05-16 08:50:24 +00:00
zhengzangw	0592723022	[feat] update bs search and loss eval	2024-05-15 13:05:53 +00:00
zhengzangw	f489cb0915	[feat] accelerate search batch size	2024-05-15 08:07:49 +00:00
zhengzangw	a36f80f847	[feat] notebook for inference	2024-05-14 07:24:56 +00:00
zhengzangw	e1e379d898	[feat] update inference config	2024-05-14 05:40:17 +00:00
zhengzangw	3a01cb440d	[feat] inference for rf	2024-05-13 11:39:16 +00:00
zhengzangw	6194da6504	[fix] update inference and datautil	2024-05-13 06:33:12 +00:00
zhengzangw	74dfaa879c	[fix] ema ckpt save	2024-05-12 05:15:06 +00:00
zhengzangw	2a55d672a7	[feat] update logging and bucket	2024-05-09 08:07:56 +00:00
zhengzangw	e9db7b3af1	update training script	2024-05-09 05:53:19 +00:00
zhengzangw	31228fe9c8	[refactor] fix train warnings	2024-05-09 03:10:26 +00:00
zhengzangw	322e566aed	update	2024-05-09 03:10:26 +00:00
Zangwei Zheng	eb5ff82fc5	quick update	2024-05-09 11:09:52 +08:00
Shen Chenhui	dc7b7fd64a	fix bug (#95) ... Co-authored-by: Shen-Chenhui <shen_chenhui@u.nus.edu>	2024-05-08 17:39:21 +08:00
Zheng Zangwei (Alex Zheng)	f047d5786a	[refactor] clean train.py code (#94)	2024-05-08 16:07:57 +08:00
Frank Lee	fe19568a49	fixed race condition in exp workspace (#91)	2024-05-07 17:17:43 +08:00
Zangwei Zheng	fea2541c70	[docs] update readme	2024-05-07 16:27:21 +08:00
Zangwei Zheng	10657621ef	[fix] random mask may have nothing to learn	2024-05-06 17:34:08 +08:00
Zheng Zangwei (Alex Zheng)	f941fa95dc	Merge branch 'dev/v1.2' into vae-clean-new	2024-05-04 20:26:42 +08:00
zhengzangw	4543bc217c	[feat] image generation	2024-05-04 10:10:54 +00:00
zhengzangw	f7a4fcfc3f	updated	2024-04-30 09:46:27 +00:00
zhengzangw	ace37cf7c7	refactor	2024-04-30 08:13:20 +00:00
zhengzangw	41e276f5ef	updated	2024-04-30 06:19:40 +00:00
周钰坤	59591d8afc	update	2024-04-30 13:36:41 +08:00
zhengzangw	1171e5b6f9	update config	2024-04-29 07:27:15 +00:00
zhengzangw	f3dcb4d6fb	complete 3b backbone	2024-04-29 06:00:14 +00:00
zhengzangw	d6a16a858e	[wip] multiple resolution	2024-04-27 11:43:15 +00:00
Shen-Chenhui	6fb4e3cd22	debug	2024-04-27 17:18:19 +08:00
YuKun Zhou	fddd08243b	Update ckpt_utils.py (#79) ... * Update ckpt_utils.py * update * Update 1x256x256.py * update * update * Update blocks.py * update * Update 1x2048x2048.py	2024-04-27 15:43:50 +08:00
Shen-Chenhui	38c46ac721	fix bug	2024-04-27 15:25:03 +08:00
zhengzangw	a4652a8aef	Merge branch 'main' into vae	2024-04-26 06:56:22 +00:00
Zangwei Zheng	e009e89e1a	add start from scratch	2024-04-23 19:24:22 +08:00
zhengzangw	8931efccd7	update config	2024-04-22 03:43:31 +00:00
Zangwei Zheng	fea6aba655	hotfix config args	2024-04-21 19:50:51 +08:00
zhengzangw	2b42ca82d8	[wip] prompt with json	2024-04-21 11:07:59 +00:00
Zheng Zangwei (Alex Zheng)	341b12f9bf	Dev/v1.0.1 (#58) ... * update (#57) * update * update datautil * add VBench prompt * update eval * update eval * update intepolation * add vbench eval * Dev/sdedit implementation (#56) * Update utils.py * update * update * update --------- Co-authored-by: YuKun Zhou <90625606+1zeryu@users.noreply.github.com>	2024-04-20 21:23:10 +08:00
zhengzangw	d262f14a24	update	2024-04-20 02:43:38 +00:00
Zangwei Zheng	74aade586a	[exp] update	2024-04-19 17:56:02 +08:00
Zangwei Zheng	15f9d702ed	[exp] update	2024-04-19 17:43:44 +08:00
Zangwei Zheng	1877a14abc	load ema strict=False	2024-04-19 16:55:03 +08:00
Shen-Chenhui	4c46972b64	debug	2024-04-19 15:59:21 +08:00
Shen-Chenhui	19187b70f5	allow specifying outputs in command line	2024-04-19 15:56:04 +08:00
Shen-Chenhui	b16001e5e3	debug	2024-04-19 14:28:53 +08:00