Skip to content

Update VAAL to incorporate recent changes in framework

Ameya Shringi requested to merge update-vaal into development

This MR updates VAAL in following ways

  1. Adds adapter to VAAL
  2. Adds python based config
  3. Create a yaml for VAAL
  4. Uses transforms that are similar to pretrain image classification
  5. Uses device along with cuda.

Currently the loss values for vae model is very high 1612180942224933122645950464.000. Not sure why this is happening on domain_net-painting.

Edited by Ameya Shringi

Merge request reports

Loading