β - vae : l earning b asic .
Adam: A Method for Stochastic Optimization