Lisa Liu-Thorrold

You might have to shuffle the d_optimzer until after `loss.backward()` has been called in the wae training loop. This was running on an older version of pytorch where the error...