Results
test_emd: 71.2318372351699
train_emd: 64.2895404644023
test_nll: 2.535527229309082
train_nll: 2.5255343914031982
