File size: 639 Bytes
d4700c7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | # The frozen autoencoder in pretrained_autoencoder.pt: the architecture every client builds, and the
# fingerprint of its weights. Written by process_tools/extract_autoencoder.py; regenerate, don't edit.
autoencoder:
spatial_dims: 3
in_channels: 1
out_channels: 1
num_res_blocks:
- 2
- 2
- 2
channels:
- 64
- 128
- 256
attention_levels:
- false
- false
- false
latent_channels: 3
norm_num_groups: 32
norm_eps: 1.0e-06
with_encoder_nonlocal_attn: false
with_decoder_nonlocal_attn: false
autoencoder_sha256: 3b5fe99af0d7fbbfe94343e24c82a200373f5d17f90c639aa9773a87b5f25a42 # pragma: allowlist secret
|