Results
test_emd: 59.76418178813578
train_emd: 62.278366161209505
test_nll: 2.540010929107666
train_nll: 2.5313284397125244
