Close

PyTorch. C++ & Python socket

A project log for NSCE-ngbrain

scalable hardware for neural network system

3drobert3drobert 11/29/2021 at 02:310 Comments

I have created a new socket on port 10000 to communicate with a python script (when PyTorch mode is enabled), to allow train&save traced Net created when inputs/outputs is received from MCU.

Now from C++ must detect the saved models and load these on the new class PTNets.

Discussions