Skip to content

Add Pytorch dependencies

millaker requested to merge millaker/unified_model into master

This change adds Pytorch dependencies using CMake and a unified model class. Currently, I assume the model takes integer inputs and outputs integers as well. If the data type is wrong, please let me know.

Furture commit will include unit test for this class. I'm thinking of loading a dummy pytorch that multiplies each input with a constant. With this test, I can ensure that the class gets expected output and works without unexpected behaviors. (The dummy model is quite easy, I've done that already)

Edited by millaker

Merge request reports