Refactoring - towards smaller models (!18) · Merge requests · gdl4hep / etx4velo · GitLab

Anthony Correia requested to merge anthonyc/smaller_models_please into dev Feb 12, 2024

New Trainings

Try n_hiddens = 32 GNN training
Explore various hyperparameters for embedding network
Try kind of incremental GNN

New Features

Refactor ONNX export in order to be able to export easily any GNN

Bug Fixing

Fix bug in get_random_pairs_plane_by_plane (embedding)