search

DeepLearnPhysics / lartpc_mlreco3d

7 stars 32 forks source link

Full chain crash fix #103

Closed dkoh0207 closed 2 years ago

dkoh0207 commented 2 years ago

This is a (temporary) crash fix for the full chain, I believe you will need both the graph-spice part and the vertex-net part.

dkoh0207 commented 2 years ago

To run GraphSPICE with/without labels, one now sets the use_cluster_labels parameter in the config as follows:

      constructor_cfg:
        mode: 'knn'
        seg_col: -1
        cluster_col: 5
        edge_mode: 'attributes'
        hyper_dimension: 22
        edge_cut_threshold: 0.1
        use_cluster_labels: False
...
    graph_spice_loss:
      name: graph_spice_edge_loss
      invert: True
      kernel_lossfn: 'lovasz_hinge'
      edge_loss_cfg:
        loss_type: 'LogDice'
      use_cluster_labels: False

If use_cluster_labels is set to True, GraphSPICE will compute edge loss/accuracy with the provided labels. Otherwise the edge loss and accuracy will be set to zero (running in data, for example).