Mentions légales du service

Skip to content
Snippets Groups Projects

Warn about frozen weights use in `TensorflowModel` and `TorchModel`.

Merged ANDREY Paul requested to merge hotfix-warn-frozen-weights into main

Related to issue #15 (closed)

This MR does not fix the aforementioned issue, that will require a minor extension to the current Model API, that will be accompanied by an extension of the current test scenarios.

Instead, it adds temporary warnings in TensorflowModel and TorchModel that are triggered when a wrapped model has frozen weights (i.e. parameters that do not require updating via gradient-descent). These warnings document the known limitations of the current support for such models (namely, that weight decay and Regularizer plug-ins cannot be used with them) and point towards the (future) version 2.1.0, that will fix those.

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
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
Please register or sign in to reply
Loading