Edge features of COSEG has nan in it. Training pipeline breaks if data["edge_feats"] is used.
Edge features of COSEG has nan in it. Training pipeline breaks if data["edge_feats"] is used.