alexandru-dinu / cae

Compressive Autoencoder.
MIT License
170 stars 31 forks source link

Explicitly control latent size (& padding etc.) #26

Open alexandru-dinu opened 3 years ago

alexandru-dinu commented 3 years ago
cae = CAE(
    latent_size=(32, 32, 32),
    padding=PADDING.zero,
)