Results
test_emd: 74.80382135489396
train_emd: 62.32336245302686
test_nll: 2.5368659496307373
train_nll: 2.525909662246704
