Commit graph

179 commits

Author SHA1 Message Date
Shen-Chenhui
b35b933841 add disc 2024-05-02 16:32:19 +08:00
zhengzangw
02904b4f29 updated 2024-05-02 03:46:01 +00:00
zhengzangw
f7a4fcfc3f updated 2024-04-30 09:46:27 +00:00
zhengzangw
b89c072eb5 [fix] micro batch size not for training 2024-04-30 09:20:08 +00:00
zhengzangw
ace37cf7c7 refactor 2024-04-30 08:13:20 +00:00
zhengzangw
41e276f5ef updated 2024-04-30 06:19:40 +00:00
zhengzangw
b2f302b379 [wip] try vae 2024-04-29 17:02:22 +00:00
Shen-Chenhui
e3b84247c1 disable t-conv for t=1 2024-04-29 20:06:18 +08:00
Shen-Chenhui
c59b499085 Merge branch 'vae-clean-new' of https://github.com/hpcaitech/Open-Sora-dev into vae-clean-new 2024-04-29 18:10:03 +08:00
Shen-Chenhui
c631358bbf add z nll loss 2024-04-29 18:09:42 +08:00
zhengzangw
9632b5bcf8 merged 2024-04-29 10:07:09 +00:00
zhengzangw
5c8139521a update 2024-04-29 10:02:42 +00:00
Shen-Chenhui
b78b469420 fix merge 2024-04-29 17:58:59 +08:00
Shen-Chenhui
f81d8648cf move out losses to different file 2024-04-29 17:53:03 +08:00
zhengzangw
a2dc903863 update 2024-04-29 09:41:39 +00:00
Shen-Chenhui
4d69671663 clean up code 2024-04-29 17:18:45 +08:00
Shen-Chenhui
98de78910d config 2024-04-29 15:21:54 +08:00
Shen-Chenhui
7811b8e99a add Nlayer 3D discriminator 2024-04-29 14:18:57 +08:00
Shen-Chenhui
069ffcc4e3 use 3 layer disc 2024-04-29 14:03:18 +08:00
Shen-Chenhui
8b465b04c5 add vae subfolder support 2024-04-29 11:32:39 +08:00
Shen-Chenhui
c9785342c2 add evals 2024-04-27 16:14:11 +08:00
Shen Chenhui
3e7ba9a46d use modulelist 2024-04-27 14:02:20 +08:00
zhengzangw
478b585024 [wip] debug vae 2024-04-26 07:27:26 +00:00
zhengzangw
a4652a8aef Merge branch 'main' into vae 2024-04-26 06:56:22 +00:00
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
Zangwei Zheng
8bf34f4418 add local_files_only 2024-04-19 16:47:26 +08:00
Shen-Chenhui
78caeea999 add support and config for z channel 4 2024-04-19 15:14: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