SyConn
SyConn copied to clipboard
Make network training functionality optional
Currently, the modules cnn and handler/prediction.py require an elektronn3 installation. Pytorch-trained networks can now be saved as jit models (.pts), which eliminates the elektronn3 dependency for handler/prediction.py once all models have been converted. This would make it possible to make the elektronn3 dependency optional since it is only needed for an activated training functionality.