78 commits

Author	SHA1	Message	Date
Shen-Chenhui	63f3737eb0	add simple discriminator	2024-04-11 15:14:16 +08:00
Shen-Chenhui	ee58ec16b4	add image encoding support	2024-04-11 10:50:23 +08:00
Shen-Chenhui	a4025b0ea7	magvit v2 enc dec arc	2024-04-09 17:49:01 +08:00
Shen-Chenhui	ae6bffb8c5	restore train vae script	2024-04-09 09:34:59 +08:00
Shen-Chenhui	a1808be460	debug	2024-04-08 18:18:05 +08:00
Shen-Chenhui	4861e6beaf	debug	2024-04-08 18:13:15 +08:00
Shen-Chenhui	9f211646f3	debug	2024-04-08 18:11:27 +08:00
Shen-Chenhui	ab38a411a8	debug	2024-04-08 18:07:50 +08:00
Shen-Chenhui	55821937d9	debug	2024-04-08 18:04:32 +08:00
Shen-Chenhui	a485925dbd	debug	2024-04-08 18:02:02 +08:00
Shen-Chenhui	45ea2bd29d	debug	2024-04-08 18:00:30 +08:00
Shen-Chenhui	e151b64319	debug	2024-04-08 17:58:33 +08:00
Shen-Chenhui	dba5de3d90	debug	2024-04-08 17:54:31 +08:00
Shen-Chenhui	bbeacb491d	debug	2024-04-08 17:50:41 +08:00
Shen-Chenhui	0d497994cd	debug	2024-04-08 17:42:27 +08:00
Shen-Chenhui	ce0b1928e6	debug	2024-04-08 17:41:08 +08:00
Shen-Chenhui	08fac69c38	debug	2024-04-08 17:39:19 +08:00
Shen-Chenhui	616e2b4fe5	debug	2024-04-08 17:37:29 +08:00
Shen-Chenhui	04a0e4769b	debug	2024-04-08 17:33:54 +08:00
Shen-Chenhui	fb64216b45	debug	2024-04-08 17:32:30 +08:00
Shen-Chenhui	25707a9d7d	debug	2024-04-08 17:28:31 +08:00
Shen-Chenhui	9c8d084ec5	debug	2024-04-08 17:26:36 +08:00
Shen-Chenhui	eae30f9f89	debug	2024-04-08 17:23:53 +08:00
Shen-Chenhui	4ef813555c	debug	2024-04-08 17:22:44 +08:00
Shen-Chenhui	827a0b2c55	debug	2024-04-08 17:21:26 +08:00
Shen-Chenhui	a634f7327e	debug	2024-04-08 17:17:22 +08:00
Shen-Chenhui	a401630e7a	debug	2024-04-08 17:03:27 +08:00
Shen-Chenhui	3eb96a2793	debug	2024-04-08 16:58:54 +08:00
Shen-Chenhui	dfae10e32b	debug	2024-04-08 16:56:28 +08:00
Shen-Chenhui	78f9ec8710	debug	2024-04-08 16:54:27 +08:00
Shen-Chenhui	47b51ba07d	debug	2024-04-08 16:53:18 +08:00
Shen-Chenhui	d927acc19d	debug	2024-04-08 16:51:06 +08:00
Shen-Chenhui	475762cdb2	debug	2024-04-08 16:37:26 +08:00
Shen-Chenhui	c6181ebdfd	debug	2024-04-08 16:36:54 +08:00
Shen-Chenhui	90fc92ee9c	debug	2024-04-08 16:14:00 +08:00
Shen-Chenhui	ab1970ca24	debug	2024-04-08 16:05:15 +08:00
Shen-Chenhui	0a5c767fc7	debug	2024-04-08 15:54:03 +08:00
Shen-Chenhui	b2148d5505	debug	2024-04-08 15:52:25 +08:00
Shen-Chenhui	7e3951300f	debug	2024-04-08 15:43:06 +08:00
Shen-Chenhui	fa0ca3983e	debug inference code	2024-04-08 15:40:14 +08:00
Shen-Chenhui	4b27448a49	debug	2024-04-08 14:22:21 +08:00
Shen-Chenhui	1b66919091	debug	2024-04-08 10:53:44 +08:00
Shen-Chenhui	ad02d28d9c	debug inference running loss	2024-04-08 10:22:07 +08:00
Shen-Chenhui	e3584b4e43	debug	2024-04-05 16:20:39 +08:00
Shen-Chenhui	9091419b28	save vae to model dir	2024-04-05 16:07:23 +08:00
Shen-Chenhui	bc095aa492	debug	2024-04-04 15:26:56 +08:00
Shen-Chenhui	dfdf8db491	add inference config ... add readme complete inference script TODO: DEBUG inference	2024-04-04 14:11:25 +08:00
Shen-Chenhui	b6c39873ee	finish discriminator arch ... use einops.rearrange in vae model_utils add get_latent_size in vae_3d inference vae_3d WIP	2024-04-03 10:29:01 +08:00
Shen-Chenhui	ebb3dc4d59	enable wandb	2024-04-02 11:47:56 +08:00
Shen-Chenhui	562a966a77	debug	2024-04-01 16:14:11 +08:00