44 commits

Author	SHA1	Message	Date
Shen-Chenhui	8d914707f8	enable end to end loss training	2024-04-27 21:46:09 +08:00
Shen-Chenhui	38c46ac721	fix bug	2024-04-27 15:25:03 +08:00
Shen-Chenhui	d7b278d4b5	removed breakpoint	2024-04-27 14:59:42 +08:00
Shen-Chenhui	2c391a636b	dataset trainscript	2024-04-27 14:55:47 +08:00
zhengzangw	478b585024	[wip] debug vae	2024-04-26 07:27:26 +00:00
Shen-Chenhui	c7a698e85b	fix dist issue	2024-04-26 13:50:56 +08:00
Shen-Chenhui	c83ee95b8d	fix bug	2024-04-26 11:57:16 +08:00
Shen-Chenhui	1d6cee302f	add lecam support	2024-04-26 11:27:20 +08:00
Shen-Chenhui	c20955e5b3	lecam support	2024-04-26 10:41:40 +08:00
Shen-Chenhui	bf1999f9b1	issue opt saving	2024-04-26 10:34:31 +08:00
Shen-Chenhui	0138be6e21	grad penalty expr	2024-04-24 14:52:47 +08:00
Shen-Chenhui	90775e52bd	debug	2024-04-24 10:12:52 +08:00
Shen-Chenhui	988bc3bb65	config lc gp	2024-04-24 10:03:58 +08:00
Shen-Chenhui	da0a875389	debug	2024-04-23 15:21:33 +08:00
Shen-Chenhui	d565477680	disable non-master print	2024-04-23 14:43:02 +08:00
Shen-Chenhui	486593f137	LeCam Loss working	2024-04-23 10:58:09 +08:00
Shen-Chenhui	8b3b96f85a	config wandb	2024-04-23 09:27:45 +08:00
Shen-Chenhui	f4fe9b5eca	add lecam	2024-04-23 09:22:05 +08:00
Shen-Chenhui	96a42e08db	config	2024-04-19 17:49:50 +08:00
Shen-Chenhui	41a600b61b	inference config	2024-04-19 10:47:09 +08:00
Shen-Chenhui	0ad337178a	debug	2024-04-19 10:29:25 +08:00
Shen-Chenhui	246eddecdd	enable pipeline vae training	2024-04-18 14:13:56 +08:00
Shen-Chenhui	aba1de6eb3	add pipeline	2024-04-18 13:49:54 +08:00
Shen-Chenhui	becff9d86f	inference v2	2024-04-16 18:15:12 +08:00
Shen-Chenhui	aae08a2841	debug	2024-04-16 17:31:57 +08:00
Shen-Chenhui	235d195dac	add logvar	2024-04-16 17:25:48 +08:00
Shen-Chenhui	36b754b047	debug	2024-04-16 15:06:17 +08:00
Shen-Chenhui	afd3f823d4	debug	2024-04-16 15:00:31 +08:00
Shen-Chenhui	af1b1e484d	reimplement blurpool	2024-04-15 17:47:33 +08:00
Shen-Chenhui	686829969f	debug	2024-04-15 15:13:29 +08:00
Shen-Chenhui	c71e04daaa	add lecam and gradient penalty loss to discriminator	2024-04-15 11:13:39 +08:00
Shen-Chenhui	79bff13099	debug	2024-04-13 10:51:56 +08:00
Shen-Chenhui	fb0f59171c	debug	2024-04-13 09:36:39 +08:00
Shen-Chenhui	06aa4589f2	debug	2024-04-12 18:39:31 +08:00
Shen-Chenhui	dbd3982ee8	debug	2024-04-12 18:37:28 +08:00
Shen-Chenhui	28f3f4b597	debug	2024-04-12 18:30:40 +08:00
Shen-Chenhui	c80459b72f	debug	2024-04-12 18:27:51 +08:00
Shen-Chenhui	f905a1b69d	debug	2024-04-12 18:16:22 +08:00
Shen-Chenhui	d53245a98c	debug	2024-04-12 18:14:18 +08:00
Shen-Chenhui	de9fffacfd	gan training	2024-04-12 17:57:35 +08:00
Shen-Chenhui	e004d43f69	gan optimizer	2024-04-12 15:10:19 +08:00
Shen-Chenhui	ab59033f20	gan	2024-04-12 12:42:02 +08:00
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