Commit graph

120 commits

Author SHA1 Message Date
Shen-Chenhui
c83ee95b8d fix bug 2024-04-26 11:57:16 +08:00
Shen-Chenhui
bf1999f9b1 issue opt saving 2024-04-26 10:34:31 +08:00
Shen-Chenhui
9683ec66c7 inference script update 2024-04-24 17:04:53 +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
eafabaee78 debug disc loss 2024-04-19 14:47:34 +08:00
Shen-Chenhui
b16001e5e3 debug 2024-04-19 14:28:53 +08:00
Shen-Chenhui
136304eb61 debug 2024-04-19 11:15:53 +08:00
Shen-Chenhui
c36d747b2f inference script 2024-04-19 11:06:13 +08:00
Shen-Chenhui
a78ffe95a6 debug 2024-04-19 10:51:11 +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
6ecc858350 pipeline config for trial data 2024-04-18 15:06:28 +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
95517d7fb5 inference v2 working 2024-04-16 18:30:31 +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
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