Results
test_emd: 68.86683967975127
train_emd: 69.82126690590917
test_nll: 2.541626214981079
train_nll: 2.531527519226074
