| {"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} |