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

config for pipeline discriminator

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-19 09:24:15 +08:00
parent 1b86c24de6
commit c28c697a0b
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
configs/vae_magvit_v2/train/pipeline_16x128x128.py
View file

@ -46,7 +46,8 @@ discriminator = dict(
num_frames = num_frames, num_frames = num_frames,
in_channels = 4, in_channels = 4,
filters = 128, filters = 128,
channel_multipliers = (2,4,4,4,4), # (2,4,4,4) for 64x64 resolution # channel_multipliers = (2,4,4,4,4), # (2,4,4,4) for 64x64 resolution
channel_multipliers= (2,4,4) # since on intermediate layer, 16 x 16 x 16 dimension z
) )