Results
test_emd: 61.21851532532182
train_emd: 66.12072140775305
test_nll: 2.536831855773926
train_nll: 2.5267555713653564
