search

brain-score / model-tools

Helper functions to extract model activations and translate from Machine Learning to Neuroscience
MIT License
8 stars 27 forks source link

Onnx wrapper #59

Open mike-ferguson opened 2 years ago

mike-ferguson commented 2 years ago

ONNX Wrapper PR - Changes still incoming, initial commit.

mschrimpf commented 2 years ago

I looked through it a bit, but stopped after realizing that the current version of this PR only supports pytorch->onnx. Let me know when tf/keras are also supported and tested