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

Support swin transformer #1

Merged
merged 10 commits into from
Jun 16, 2021
Merged

Support swin transformer #1

merged 10 commits into from
Jun 16, 2021

Conversation

ZwwWayne
Copy link
Collaborator

@ZwwWayne ZwwWayne commented May 24, 2021

This PR adds Swin Transformer as an example to show how we can use MIM to run Swin Transformer for object detection and segmentation with only two python file and several configs.

Copy link
Member

@kennymckormick kennymckormick left a comment

Choose a reason for hiding this comment

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

Besides, we want to keep the examples as simple as possible. Can we remove all those base configs and merge them into the configs actually used? So that we will have much fewer files in this example.

swin_transformer/slurm_train.sh Outdated Show resolved Hide resolved
swin_transformer/README.md Outdated Show resolved Hide resolved
@kennymckormick
Copy link
Member

@ZwwWayne Do you have some time to push this PR forward. I think it can be merged after some reorg (to reduce file numbers) and adding a requirements.txt. I think this PR will be a milestone of MIM-Examples.

@ZwwWayne
Copy link
Collaborator Author

ZwwWayne commented Jun 8, 2021

@ZwwWayne Do you have some time to push this PR forward. I think it can be merged after some reorg (to reduce file numbers) and adding a requirements.txt. I think this PR will be a milestone of MIM-Examples.

sure, will make it by this week

@ZwwWayne
Copy link
Collaborator Author

ZwwWayne commented Jun 8, 2021

@kennymckormick Can have a look. Now, this PR works with MMDetection. I cannot run mmseg due to some reasons in mim. Will check that tomorrow.

@kennymckormick
Copy link
Member

Hi, @ZwwWayne , will you support Swin + segmentation in this PR (I think detection only is OK)? Besides, would you please also provide a Swin checkpoint so that users can run testing in this MIM example.

@ZwwWayne ZwwWayne merged commit fa4a200 into open-mmlab:master Jun 16, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants