Results
test_emd: 57.63657561742955
train_emd: 65.47016034577335
test_nll: 2.540666103363037
train_nll: 2.529728651046753
