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.

Sign up for GitHub

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

Jump to bottom

Dino model #74

Open
wants to merge 9 commits into
base: main
Choose a base branch
from
Open

Dino model #74

wants to merge 9 commits into from

Conversation

OmarJabri7
Copy link

Finalized DINO model architecture split into three main files:

  • dino.py containing the DINONet and Spec classes
  • layers.py containing the inner layers that the DINI model builds on top: DINO Back Bone ( vision Transfomer) and DINO head (projection head)
  • utils.py containing utility functions that the different DINO modules use

and a testing script under test_dinonet.py, which displays the DINO model in action against image data.

@a0m0rajab
Copy link

@OmarJabri7 That's a great work you did, can you message me when this is merged so that we can have a blog post draft about it?

@OmarJabri7
dinoadded test dino net as well fix to vit with ivy.expand instead of…
b52a479 
… expand_dims \nNeed to fix DINO net with nlayers > 1, currently nlayers are 1 and it is working fine
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@OmarJabri7 @a0m0rajab