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 models for image classification task #175

Conversation

wuzijian1997
Copy link

  1. Add torchvision_classif.py to models. This file integrated ResNet, DenseNet, Vision Transformer, Swin Transformer, and Swin Transformer V2 based on the Torchvision library.
  2. Delete the previous resnet.py

Copy link
Collaborator

@zzachw zzachw left a comment

Choose a reason for hiding this comment

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

LGTM. Thanks Zijian for the PR!

@zzachw zzachw merged commit 52f9d15 into sunlabuiuc:zhenbang/f-image_text_support Jun 23, 2023
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.

None yet

2 participants