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

Bigger ResNets #11

Open
wants to merge 4 commits into
base: master
Choose a base branch
from
Open

Bigger ResNets #11

wants to merge 4 commits into from

Conversation

adjavon
Copy link
Contributor

@adjavon adjavon commented Sep 20, 2024 •
edited
Loading

Current ResNet model is only ResNet18, which is much smaller than the Vgg and therefore probably not a fair comparison.
It also splits ResNet 2d and 3d, with a large amount of overlap.

This PR modifies the code to:

  • Merge ResNet 2d and 3d into a single class (with convenience classes added in, for backwards compatibility)
  • Add ResNet34
  • Add ResNet50
  • Add ResNet101
  • Add ResNet152

@adjavon adjavon self-assigned this Sep 20, 2024
@adjavon
Generalize the versions
001ac91 
Instead of specifically defining version 18, 34, we can just define the
number of repeats for each set of blocks.
@adjavon
Add bottleneck block, for bigger models.
0d66037
@adjavon adjavon marked this pull request as ready for review September 25, 2024 21:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@pattonw pattonw Awaiting requested review from pattonw

Assignees

@adjavon adjavon

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@adjavon