neoml
neoml copied to clipboard
Machine learning framework for both deep learning and traditional algorithms
In most cases .onnx models are produced from Python environment - e.g. from `tensorflow + tf2onnx` or `torch + torch.onnx.export`). Currently to convert such exported onnx models into NeoML you...
Whenever I try to build NeoML/Python with `python setup.py install`, I get the following error: ``` ... -- Configuring done CMake Error in CMakeLists.txt: CUDA_ARCHITECTURES is empty for target "PythonWrapper"....
It will be nice to have general functor in CLayerWrapper without restriction in inputs amount. It wil be useful for layers like Concat etc. General functor sugestion: ``` // Connects...
If some weights in the network become NaN at some point during training, I'd like the training to stop with an error. Currently, when training on a GPU, there is...