Scan over strength of residual connections

gnn-tracking / gnn_tracking

Reconstruct billions of particle trajectories with graph neural networks

https://gnn-tracking.rtfd.io/

MIT License

30 stars 13 forks source link

Scan over strength of residual connections #507

Open klieret opened 3 months ago

klieret commented 3 months ago

In the track condensation network, scan over the strength of the residual connection alpha_hc (currently set to 0.5 by default). Example config:

class_path: gnn_tracking.training.tc.TCModule
init_args:
  model:
    class_path: gnn_tracking.models.track_condensation_networks.GraphTCNForMLGCPipeline
    init_args:
      alpha_hc: 0.5