Skip to content

Add new weights for wide_resnet50_2 model #4862

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

Merged

Conversation

prabhat00155
Copy link
Contributor

@prabhat00155 prabhat00155 commented Nov 4, 2021

Fixes partially #3995

torchrun --nproc_per_node=1 train.py --test-only --weights ImageNet1K_RefV2 --model wide_resnet50_2 -b 1
Test:  Acc@1 81.602 Acc@5 95.758

cc @datumbox @bjuncek

@facebook-github-bot
Copy link

facebook-github-bot commented Nov 4, 2021

💊 CI failures summary and remediations

As of commit 2341bad (more details on the Dr. CI page):


  • 2/2 failures introduced in this PR

2 failures not recognized by patterns:

Job Step Action
CircleCI binary_linux_conda_py3.7_cu111 packaging/build_conda.sh 🔁 rerun
CircleCI binary_libtorchvision_ops_android Build 🔁 rerun

1 job timed out:

  • binary_linux_conda_py3.7_cu111

This comment was automatically generated by Dr. CI (expand for details).

Please report bugs/suggestions to the (internal) Dr. CI Users group.

Click here to manually regenerate this comment.

@datumbox datumbox changed the title Updated wide_resnet50_2 model weights Add new weights for wide_resnet50_2 model Nov 4, 2021
Copy link
Contributor

@datumbox datumbox left a comment

Choose a reason for hiding this comment

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

@prabhat00155 Awesome result Prabhat! Looks good to me.

Feel free to merge once the accuracy is estimated on 1 GPU and 1 image per batch.

@prabhat00155 prabhat00155 merged commit ed2eefc into pytorch:main Nov 5, 2021
@prabhat00155 prabhat00155 deleted the prabhat00155/wide_resnet_update branch November 5, 2021 13:21
facebook-github-bot pushed a commit that referenced this pull request Nov 8, 2021
Summary:
* Updated wide_resnet50_2 model weights

* Update Accs with batch-size 1.

* Update checkpoint and accs.

Reviewed By: kazhang

Differential Revision: D32216677

fbshipit-source-id: 3ea09f12d85f88c131bd29ac8482ae420c17d10f

Co-authored-by: Vasilis Vryniotis <[email protected]>
cyyever pushed a commit to cyyever/vision that referenced this pull request Nov 16, 2021
* Updated wide_resnet50_2 model weights

* Update Accs with batch-size 1.

* Update checkpoint and accs.

Co-authored-by: Vasilis Vryniotis <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Improve the accuracy of Classification models by using SOTA recipes and primitives
3 participants