Skip to content

Add pretrained weights on ImageNet for some models#197

Merged
darsnack merged 6 commits intoFluxML:masterfrom
theabhirath:weights
Aug 10, 2022
Merged

Add pretrained weights on ImageNet for some models#197
darsnack merged 6 commits intoFluxML:masterfrom
theabhirath:weights

Conversation

@theabhirath
Copy link
Member

@theabhirath theabhirath commented Aug 4, 2022

This PR will add pretrained weights on ImageNet ported from torchvision for some models like ResNet, WideResNet, ResNeXt and SqueezeNet.

  • Merge HF PRs for new weights
  • Update Artifacts.toml

@theabhirath theabhirath added this to the 0.8 milestone Aug 4, 2022
@theabhirath theabhirath added the model-weights Related to pretrained weights for models label Aug 4, 2022
@theabhirath theabhirath marked this pull request as ready for review August 10, 2022 15:54
@theabhirath theabhirath requested a review from darsnack August 10, 2022 15:54
Copy link
Member

@darsnack darsnack left a comment

Choose a reason for hiding this comment

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

Just update the README table?

@darsnack darsnack merged commit 1e4c669 into FluxML:master Aug 10, 2022
@theabhirath theabhirath deleted the weights branch August 10, 2022 16:44
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

model-weights Related to pretrained weights for models

Projects

No open projects

Development

Successfully merging this pull request may close these issues.

2 participants