Results
test_emd: 56.13850089824924
train_emd: 62.04835505561178
test_nll: 2.5315959453582764
train_nll: 2.522152900695801
