Skip to content

Conversation

sachinprasadhs
Copy link
Collaborator

@sachinprasadhs sachinprasadhs commented Feb 11, 2025

  • Reimplement the CspNet architecture from CspDarkNet to be more inclined towards timm implementation, which will enable us to port many models under CspNet.
  • Timm conversion script for 'cspdarknet53.ra_in1k'
  • Added test cases for timm conversion, backbone test, classifier test.

Attaching the colab notebook with the demonstration of conversion and the model output.

Copy link
Collaborator

@divyashreepathihalli divyashreepathihalli left a comment

Choose a reason for hiding this comment

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

Thanks @sachinprasadhs !! did a high level pass at the PR

Copy link
Collaborator

@divyashreepathihalli divyashreepathihalli left a comment

Choose a reason for hiding this comment

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

waiting for preset upload and preset file update. LGTM otherwise.

@sachinprasadhs sachinprasadhs added the kokoro:force-run Runs Tests on GPU label Mar 4, 2025
@kokoro-team kokoro-team removed the kokoro:force-run Runs Tests on GPU label Mar 4, 2025
@sachinprasadhs sachinprasadhs added the kokoro:force-run Runs Tests on GPU label Mar 4, 2025
@kokoro-team kokoro-team removed the kokoro:force-run Runs Tests on GPU label Mar 4, 2025
@sachinprasadhs sachinprasadhs added the kokoro:force-run Runs Tests on GPU label Mar 5, 2025
@kokoro-team kokoro-team removed the kokoro:force-run Runs Tests on GPU label Mar 5, 2025
@divyashreepathihalli divyashreepathihalli merged commit 0cbaf0f into keras-team:master Mar 6, 2025
10 checks passed
@sachinprasadhs sachinprasadhs deleted the cspnet branch March 6, 2025 18:15
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