24 commits

Author	SHA1	Message	Date
Shen-Chenhui	63f3737eb0	add simple discriminator	2024-04-11 15:14:16 +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	aa733aa1d7	debug	2024-04-02 11:21:01 +08:00
Shen-Chenhui	562a966a77	debug	2024-04-01 16:14:11 +08:00
Shen-Chenhui	6bed1bdd0b	working perceptual loss	2024-04-01 16:11:21 +08:00
Shen-Chenhui	6f460c1d05	debug	2024-04-01 16:06:47 +08:00
Shen-Chenhui	036b427b00	debug	2024-04-01 15:38:29 +08:00
Shen-Chenhui	996e6fb180	debug	2024-04-01 15:33:39 +08:00
Shen-Chenhui	49498b7cdb	debug	2024-04-01 15:29:08 +08:00
Shen-Chenhui	85f929d126	debug	2024-04-01 15:26:16 +08:00
Shen-Chenhui	c687b700bc	debug	2024-04-01 14:53:11 +08:00
Shen-Chenhui	eda89dd643	debug	2024-04-01 14:25:37 +08:00
Shen-Chenhui	dbda9e9c3a	debug	2024-04-01 14:18:58 +08:00
Shen-Chenhui	3d6d499543	add perceptual loss	2024-04-01 13:57:25 +08:00
Shen-Chenhui	60a0a6ea8f	clean up vae (rec+kl loss)	2024-03-29 10:43:33 +08:00
Shen-Chenhui	1f81907ceb	debug	2024-03-29 09:59:07 +08:00
Shen-Chenhui	3aa248d7cf	debug	2024-03-29 09:31:43 +08:00
shenchenhui	d927619833	debug	2024-03-28 23:08:32 +08:00
shenchenhui	59d36321c9	debug	2024-03-28 22:58:26 +08:00
shenchenhui	b3f2dacc69	debug	2024-03-28 16:44:41 +08:00
shenchenhui	383ae23859	debug	2024-03-28 16:39:50 +08:00
shenchenhui	afff661efd	debug	2024-03-28 16:22:33 +08:00
shenchenhui	a5917c6a3e	added vae3d training code	2024-03-28 15:12:20 +08:00