Emd training merge

[stsan9]

Done:

• reuse code for graph_data between emd-training and gae repos
• relative pt eta phi
• adapt emd losses to use new data format
• sub dirs

WIP:

• option for old preprocessing (px, py, pz)
• add in rest of emd-training code
• testing