BWN133

Thanks a lot for replying!!! My original configs looks like the following: ``` model_params: name: 'VanillaVAE' in_channels: 3 latent_dim: 128 data_params: data_path: "Data/" train_batch_size: 64 val_batch_size: 64 patch_size: 64 num_workers:...