nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-11 05:13:31 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

lecam config

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-23 09:59:12 +08:00
parent 8b3b96f85a
commit 16b0149fc4
2 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
configs/vae_magvit_v2/train/16x128x128_z4_LC.py
View file

@ -55,7 +55,7 @@ generator_factor = 0.1 # SCH: generator adversarial loss, MAGVIT v2 uses 0.1
lecam_loss_weight = 0.001 # NOTE: MAVGIT v2 use 0.001
discriminator_loss_type="non-saturating"
generator_loss_type="non-saturating"
discriminator_start = 1000 # 50000 NOTE: change to correct val, debug use -1 for now
discriminator_start = -1 # 50000 NOTE: change to correct val, debug use -1 for now
gradient_penalty_loss_weight = None # 10 # SCH: MAGVIT uses 10, opensora plan doesn't use
ema_decay = 0.999 # ema decay for LeCam

6
opensora/models/vae/README.md
View file

@ -1,5 +1,11 @@
## Commands
## 0. References
* https://github.com/google-research/magvit
* https://github.com/CompVis/taming-transformers
* https://github.com/adobe/antialiased-cnns/pull/39/commits/3d6f02b6943c58b68c19c07bc26fad57492ff3bc
## 1. VAE 3D
### 1.1 Train