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

[ResNet] Upgrade to Burn 0.13.0 #25

Merged
merged 5 commits into from
Apr 16, 2024
Merged

[ResNet] Upgrade to Burn 0.13.0 #25

merged 5 commits into from
Apr 16, 2024

Conversation

laggui
Copy link
Member

@laggui laggui commented Apr 15, 2024

Closes #19

  • Updated burn and image versions
  • Changed ReLU -> Relu
  • Removed init_with methods (yay lazy init)
  • Refactored residual blocks to use newly supported enum modules
  • Changes burn-rs -> tracel-ai github links
  • Added with_classes method to support output layer initialization for fine-tuning

@laggui
Copy link
Member Author

laggui commented Apr 15, 2024

Forgot I don't need a review approval on the models repo to merge.. will still wait for at least one review to merge for sanity :)

Copy link
Collaborator

@antimora antimora left a comment

Choose a reason for hiding this comment

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

LGTM

@laggui laggui merged commit 5677dc0 into main Apr 16, 2024
2 checks passed
@laggui laggui deleted the chore/upgrade-burn branch April 16, 2024 12:05
@laggui laggui mentioned this pull request Apr 17, 2024
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.

[ResNet] Chore: Use Candle's published crate version instead of git revision
2 participants