Aidge_learning tutorial - cannot create an SGD optimizer (CPU)
#Aidge tutorial - Learning
Environnement
- aidge 0.4.0
- aidge_backend_cpu 0.4.0
- aidge_backend_cuda 0.4.0
- aidge_core 0.4.0
- aidge_learning 0.2.2
- aidge_onnx 0.4.0
Train a model on CPU
The goal of the tutorial is to train a simple model on a specific backend. Here I chose the one given in the tutorial : BACKEND = 'cpu'
I have an error in the tutorial in the part where I setup the learning objects :
Cell :
opt = aidge_learning.SGD()
learning_rates = aidge_learning.constant_lr(0.01)
opt.set_learning_rate_scheduler(learning_rates)
opt.set_parameters(list(aidge_core.producers(model)))
Error:
Assertion failed: it != C::registry().cend() in .venv_aidge/lib/libAidge/include/aidge/utils/Registrar.hpp:80
missing or invalid registrar key: ("cpu", "Float32") for registrable object N5Aidge6TensorE
Did you include/import the corresponding module?
If so, it is possible that the object is not yet supported.
Problem
I cannot create on optimizer for the training on CPU. Where does this error come from? An environnement incompatibility or something else?
Edited by Alice Batte