Results
test_emd: 61.81468003320834
train_emd: 58.14436483927011
test_nll: 2.542708158493042
train_nll: 2.5334672927856445
