Skip to content

Feature/upgrade timm mlflow onnxconverter#155

Merged
michele-milesi merged 10 commits intomainfrom
feature/upgrade-timm-mlflow-onnxconverter
Feb 13, 2026
Merged

Feature/upgrade timm mlflow onnxconverter#155
michele-milesi merged 10 commits intomainfrom
feature/upgrade-timm-mlflow-onnxconverter

Conversation

@michele-milesi
Copy link
Member

@michele-milesi michele-milesi commented Feb 13, 2026

Summary

Describe the purpose of the pull request, including:

  • Upgraded timm
  • Upgraded onnxconverter-common
  • Upgraded mlflow
  • Added backbones
  • Fixed properties initialization of SklearnMlflowClient

Type of Change

Please select the one relevant option below:

  • New feature (non-breaking change that adds functionality)

Checklist

Please confirm that the following tasks have been completed:

  • I have tested my changes locally and they work as expected. (Please describe the tests you performed.)
  • I have added unit tests for my changes, or updated existing tests if necessary.
  • I have updated the documentation, if applicable.
  • I have installed pre-commit and run locally for my code changes.

Copy link

Copilot AI left a comment

Choose a reason for hiding this comment

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

Pull request overview

This PR updates core ML/experiment dependencies (timm, mlflow-skinny, onnxconverter-common), bumps the library version to 2.8.0, and expands the available timm backbone Hydra configs while adjusting a few code paths for the upgraded libraries (notably timm imports and MLflow logging/client initialization).

Changes:

  • Upgrade dependencies and bump quadra version to 2.8.0 (incl. lockfile refresh).
  • Add many new backbone configuration YAMLs and update a couple of existing backbone model names.
  • Adjust timm imports and update MLflow logging/client initialization behavior.

Reviewed changes

Copilot reviewed 36 out of 37 changed files in this pull request and generated 3 comments.

Show a summary per file
File Description
quadra/utils/models.py Update timm import path for DropPath; tweak typing/casts around gradcam paths.
quadra/utils/mlflow.py Change how models are logged to MLflow (switch to name=..., add step=-1) and initialize MLflow client properties earlier.
quadra/models/classification/backbones.py Update timm import path for load_checkpoint.
quadra/configs/backbone/csatv2.yaml New timm backbone config.
quadra/configs/backbone/dinov3_vitb16.yaml New timm backbone config.
quadra/configs/backbone/dinov3_vits16.yaml New timm backbone config.
quadra/configs/backbone/edgenext_s.yaml New timm backbone config.
quadra/configs/backbone/efficientnet_em.yaml New timm backbone config.
quadra/configs/backbone/efficientnetv2_b1.yaml New timm backbone config.
quadra/configs/backbone/fasternet_t0.yaml New timm backbone config.
quadra/configs/backbone/fasternet_t1.yaml New timm backbone config.
quadra/configs/backbone/fasternet_t2.yaml New timm backbone config.
quadra/configs/backbone/ghostnet_v3.yaml New timm backbone config.
quadra/configs/backbone/hgnet_v2.yaml New timm backbone config.
quadra/configs/backbone/mobilenet_edgetpu_v2.yaml New timm backbone config.
quadra/configs/backbone/mobilenetv4_m.yaml New timm backbone config.
quadra/configs/backbone/mobilenetv4_s.yaml New timm backbone config.
quadra/configs/backbone/regnety_016.yaml New timm backbone config.
quadra/configs/backbone/repvit.yaml New timm backbone config.
quadra/configs/backbone/rexnet.yaml New timm backbone config.
quadra/configs/backbone/shvit_s1.yaml New timm backbone config.
quadra/configs/backbone/shvit_s2.yaml New timm backbone config.
quadra/configs/backbone/shvit_s3.yaml New timm backbone config.
quadra/configs/backbone/shvit_s4.yaml New timm backbone config.
quadra/configs/backbone/starnet_s2.yaml New timm backbone config.
quadra/configs/backbone/starnet_s3.yaml New timm backbone config.
quadra/configs/backbone/starnet_s4.yaml New timm backbone config.
quadra/configs/backbone/swiftformer_l1.yaml New timm backbone config.
quadra/configs/backbone/swiftformer_l3.yaml New timm backbone config.
quadra/configs/backbone/swiftformer_s.yaml New timm backbone config.
quadra/configs/backbone/swiftformer_xs.yaml New timm backbone config.
quadra/configs/backbone/resnet18_ssl.yaml Update timm model name for SSL ResNet18.
quadra/configs/backbone/efficientnetv2_s.yaml Update timm model name for EfficientNetV2-S.
quadra/init.py Bump package version constant to 2.8.0.
pyproject.toml Bump project version and pin upgraded dependencies (mlflow-skinny/timm/onnxconverter-common).
poetry.lock Lockfile refresh for upgraded dependency set.
CHANGELOG.md Add 2.8.0 release notes (new backbones + dependency upgrades + MLflow client fix).

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copy link
Contributor

@lorenzomammana lorenzomammana left a comment

Choose a reason for hiding this comment

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

Nice

@michele-milesi michele-milesi merged commit 24bc19f into main Feb 13, 2026
1 of 2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants