Results
test_emd: 79.61672874756114
train_emd: 80.28637550840193
test_nll: 2.5453402996063232
train_nll: 2.5346925258636475
