Skip to content
New issue

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

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

Already on GitHub? # to your account

Migrate ML model to PyTorch #679

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

Dmitriy-NK
Copy link
Contributor

Description

Keras to PyTorch migration

  • Replace model architecture and model train scripts with PyTorch-based ones
  • Train model with updated scripts
  • Implement hyperparameter search with Optuna
  • Refactor ML model metrics history
  • Remove TensorFlow dependencies

How has this been tested?

Please describe the tests that you ran to verify your changes.

  • UnitTest
  • Benchmark

@Dmitriy-NK Dmitriy-NK added the enhancement New feature or request label Feb 17, 2025
@Dmitriy-NK Dmitriy-NK self-assigned this Feb 17, 2025
@codecov-commenter
Copy link

codecov-commenter commented Feb 17, 2025

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 92.51%. Comparing base (54b25b1) to head (4ac14c7).

Additional details and impacted files
@@            Coverage Diff             @@
##             main     #679      +/-   ##
==========================================
- Coverage   92.53%   92.51%   -0.02%     
==========================================
  Files         137      137              
  Lines        5049     5049              
  Branches      828      828              
==========================================
- Hits         4672     4671       -1     
  Misses        229      229              
- Partials      148      149       +1     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@Dmitriy-NK Dmitriy-NK force-pushed the pytorch_migration branch 3 times, most recently from 42de925 to 8880657 Compare February 17, 2025 16:05
@Dmitriy-NK Dmitriy-NK marked this pull request as ready for review February 17, 2025 16:33
@Dmitriy-NK Dmitriy-NK requested a review from a team as a code owner February 17, 2025 16:33
@Dmitriy-NK Dmitriy-NK force-pushed the pytorch_migration branch 5 times, most recently from 7d86af4 to 24ec556 Compare February 21, 2025 17:47
@Dmitriy-NK Dmitriy-NK force-pushed the pytorch_migration branch 4 times, most recently from 3bb4a11 to 4f044ff Compare February 27, 2025 17:10
@babenek
Copy link
Contributor

babenek commented Feb 28, 2025

Thank you for the experiment!

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants