Results
test_emd: 71.99860981414939
train_emd: 73.04939413075
test_nll: 2.538508892059326
train_nll: 2.5269155502319336
