For implementing conditional VAE, we need auxiliary loss to ensure that the output is conditioned on the given class label (c).
In order to do that, we need to modify entire learning pipeline, including dataloader, criteria intialization and usage, gradient update, and additional modules to implement auxiliary loss.
Aforementioned stuffs will be implemented under dev_uyoung branch.