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 |
|