Fix tensor data type in optimizer (SGD, Adam) (!42) · Merge requests · Eclipse Projects / aidge / aidge_learning · GitLab

This is an archived project. Repository and other project resources are read-only.

Olivier Antoni requested to merge oantoni/aidge_learning:fix_optim_dataType into dev Apr 14, 2025

Context

It is not possible to use an Optimizer (SGD, Adam) for a data type different from float32.

Detailed major modifications

Use the setDataType method for all tensors involved in Optimizer.

Copyright © Eclipse Foundation, Inc. All Rights Reserved. Privacy Policy | Terms of Use | Copyright Agent