Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix Learning Rate and Loss Handling in Tile Classifier MaskRCNN EfficientNet #3873

Conversation

eugene123tw
Copy link
Contributor

Summary

  • Custom Learning Rate Adjustment: Fixed an issue with the customized learning rate for MaskRCNN EfficientNet that caused 0% accuracy when the tile classifier was enabled.
  • Loss Dictionary Update: stable training by removing "acc" from loss_dict to prevent gradient explosion

How to test

Checklist

  • I have added unit tests to cover my changes.​
  • I have added integration tests to cover my changes.​
  • I have ran e2e tests and there is no issues.
  • I have added the description of my changes into CHANGELOG in my target branch (e.g., CHANGELOG in develop).​
  • I have updated the documentation in my target branch accordingly (e.g., documentation in develop).
  • I have linked related issues.

License

  • I submit my code changes under the same Apache License that covers the project.
    Feel free to contact the maintainers if that's a concern.
  • I have updated the license header for each file (see an example below).
# Copyright (C) 2024 Intel Corporation
# SPDX-License-Identifier: Apache-2.0

@eugene123tw eugene123tw requested a review from a team as a code owner August 21, 2024 09:27
@github-actions github-actions bot added the ALGO Any changes in OTX Algo Tasks implementation label Aug 21, 2024
Copy link
Contributor

@chuneuny-emily chuneuny-emily left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can you add unit test code for this case?

@github-actions github-actions bot added the TEST Any changes in tests label Aug 21, 2024
@chuneuny-emily chuneuny-emily merged commit 1fced5a into openvinotoolkit:releases/1.6.0 Aug 23, 2024
13 of 14 checks passed
@eugene123tw eugene123tw deleted the eugene/fix-tile-classifier-optimization branch September 12, 2024 10:34
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ALGO Any changes in OTX Algo Tasks implementation TEST Any changes in tests
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants