Results
test_emd: 67.52882565638123
train_emd: 66.8150519776245
test_nll: 2.5334553718566895
train_nll: 2.5243961811065674
