multitask-ae / config.json
Aubins's picture
Upload multitask autoencoder model
a442f2e verified
{
"activation": null,
"in_channels": 3,
"input_size": 32,
"latent_dim": 128,
"num_classes": 10,
"out_channels": 64
}