vae-landscape-128 / config.json
bhavya777's picture
Initial upload of trained VAE model
c9a8a6e verified
raw
history blame contribute delete
282 Bytes
{"block_out_channel": [64, 64, 128, 256], "input_channel": 3, "output_channel": 3, "latent_channel": 8, "num_res_layers": 2, "group_channels": 16, "lr": 0.0001, "beta": 0.1, "epochs": 10, "batch": 16, "image_size": 128, "folder_path": "/kaggle/working/vae_data", "train_split": 0.9}