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

Add example to fine-tune TorchVision models using Keras #909

Merged
merged 4 commits into from
Sep 18, 2023

Conversation

soumik12345
Copy link
Contributor

This PR adds an example to fine-tune pre-trained models from TorchVision using KerasCore.

Notebook: https://colab.research.google.com/drive/14CkRocZrvGxFCn6-zOaFC_gji35nF-Fc?usp=sharing

Linked Issue: keras-team/keras#18406

@codecov
Copy link

codecov bot commented Sep 18, 2023

Codecov Report

Patch and project coverage have no change.

Comparison is base (a465816) 76.82% compared to head (92a302f) 76.82%.

Additional details and impacted files
@@           Coverage Diff           @@
##             main     keras-team/keras-core#909   +/-   ##
=======================================
  Coverage   76.82%   76.82%           
=======================================
  Files         329      329           
  Lines       31427    31427           
  Branches     6112     6112           
=======================================
  Hits        24144    24144           
  Misses       5719     5719           
  Partials     1564     1564           
Flag Coverage Δ
keras_core 76.73% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

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

Copy link
Member

@fchollet fchollet left a comment

Choose a reason for hiding this comment

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

Thanks for the PR!

examples/keras_io/pytorch/torchvision_keras.py Outdated Show resolved Hide resolved
examples/keras_io/pytorch/torchvision_keras.py Outdated Show resolved Hide resolved
examples/keras_io/pytorch/torchvision_keras.py Outdated Show resolved Hide resolved
Copy link
Member

@fchollet fchollet left a comment

Choose a reason for hiding this comment

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

Thanks for the contribution! LGTM

@fchollet fchollet merged commit 9d65850 into keras-team:main Sep 18, 2023
8 checks passed
@soumik12345 soumik12345 deleted the example/torchvision-keras branch September 19, 2023 07:26
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.

2 participants