swim_new / configs /model /autoencoder.yaml
qninhdt's picture
cc
8cc0674
raw
history blame
193 Bytes
_target_: swim.models.autoencoder.Autoencoder
learning_rate: 1e-4
channels: 128
channel_multipliers: [1, 2, 4]
n_resnet_blocks: 1
in_channels: 3
out_channels: 3
z_channels: 4
emb_channels: 4