Commit graph

178 commits

Author SHA1 Message Date
Shen-Chenhui
576b44d98e add lecam storage support 2024-04-26 10:34:00 +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
988bc3bb65 config lc gp 2024-04-24 10:03:58 +08:00
Shen-Chenhui
48e3a84ba6 enable r1 gradient penalty 2024-04-23 17:51:50 +08:00
Shen-Chenhui
d565477680 disable non-master print 2024-04-23 14:43:02 +08:00
Shen-Chenhui
d0d7da36ea silence lpips 2024-04-23 11:44:33 +08:00
Shen-Chenhui
2ac952bb71 8 GPU training config 2024-04-23 11:30:40 +08:00
Shen-Chenhui
486593f137 LeCam Loss working 2024-04-23 10:58:09 +08:00
Shen-Chenhui
16b0149fc4 lecam config 2024-04-23 09:59:12 +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
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
78caeea999 add support and config for z channel 4 2024-04-19 15:14:53 +08:00
Shen-Chenhui
b16001e5e3 debug 2024-04-19 14:28:53 +08:00
Shen-Chenhui
89a58b8e15 debug 2024-04-19 14:23:57 +08:00
Shen-Chenhui
1c3e910fc6 allow loading of discriminator in inference 2024-04-19 14:19:29 +08:00
Shen-Chenhui
6581556997 pipeline config 2024-04-19 13:36:47 +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
34be856fdc add disc initialization code 2024-04-19 09:39:23 +08:00
Shen-Chenhui
1b86c24de6 remove breakpoint 2024-04-18 15:22:22 +08:00
Shen-Chenhui
60892cd090 debug 2024-04-18 15:11:25 +08:00
Shen-Chenhui
246eddecdd enable pipeline vae training 2024-04-18 14:13:56 +08:00
Shen-Chenhui
f867ebc819 debug 2024-04-18 13:29:59 +08:00
Shen-Chenhui
8bf4a0fa77 add disable space in vae v2 2024-04-18 12:23:02 +08:00
Shen-Chenhui
e241ecbbe1 use default discriminator 2024-04-18 09:32:17 +08:00
Shen-Chenhui
5084d4e4f9 disc 2024-04-17 18:12:56 +08:00
Shen-Chenhui
de0199c6b2 debug 2024-04-16 17:30:03 +08:00
Shen-Chenhui
235d195dac add logvar 2024-04-16 17:25:48 +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
8aad3d6de2 debug 2024-04-12 18:35:28 +08:00
Shen-Chenhui
c3763cb556 debug 2024-04-12 18:32:24 +08:00
Shen-Chenhui
5c046c4771 debug 2024-04-12 18:22:35 +08:00
Shen-Chenhui
735e9e722b debug 2024-04-12 18:18:06 +08:00
Shen-Chenhui
d53245a98c debug 2024-04-12 18:14:18 +08:00
Shen-Chenhui
17d6b58178 debug 2024-04-12 18:08:42 +08:00
Shen-Chenhui
eae76b5a04 debug 2024-04-12 18:00:52 +08:00
Shen-Chenhui
de9fffacfd gan training 2024-04-12 17:57:35 +08:00
Shen-Chenhui
dfb6af84d4 debug 2024-04-12 15:21:40 +08:00
Shen-Chenhui
a97c70fab2 debug 2024-04-12 15:20:50 +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
19fb131e49 debug 2024-04-11 17:54:11 +08:00