Add device argument to model restoring

gnn-tracking / gnn_tracking

Reconstruct billions of particle trajectories with graph neural networks

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

MIT License

33 stars 17 forks source link

Closed klieret closed 1 year ago

codecov[bot] commented 1 year ago

Patch has no changes to coverable lines.

Files Changed	Coverage
...nn_tracking/graph_construction/data_transformer.py	`0.00%`
src/gnn_tracking/models/edge_classifier.py	`0.00%`
src/gnn_tracking/utils/lightning.py	`0.00%`

:loudspeaker: Thoughts on this report? Let us know!.