GNN_Tracking_Task1 To train the data: python3 main.py To train data with unbalanced loss function: python3 main.py balanced