Skip to content
Snippets Groups Projects

Fix tensor data type in optimizer (SGD, Adam)

Merged Olivier Antoni requested to merge oantoni/aidge_learning:fix_optim_dataType into dev

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.

Merge request reports

Loading
Loading

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
Please register or sign in to reply
Loading