Skip to content

UNet model in pytorch #79

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
Merged
merged 43 commits into from
Feb 5, 2022
Merged

UNet model in pytorch #79

merged 43 commits into from
Feb 5, 2022

Conversation

AhmetCanSolak
Copy link
Collaborator

@AhmetCanSolak AhmetCanSolak commented Jan 11, 2022 •
edited
Loading

This PR introduces pytorch implementation of our current UNet model in use. Idea is eventually having everything in nn submodule implemented in pytorch and migrate totally from tensorflow/keras.

@AhmetCanSolak AhmetCanSolak self-assigned this Jan 11, 2022
@AhmetCanSolak
introducing initial test cases for the torch unet model
7524071
@AhmetCanSolak
forward method template implemented, details will be completed and te…
cebfe0f 
…st before merging the PR, currently only provides a template implementation
@AhmetCanSolak
black fixes
1322b53
@AhmetCanSolak
import fix
4666c3f
@AhmetCanSolak
import fix on custom_rot_90
025d993
@AhmetCanSolak
implemented ConvWithBatchNorm, missing some module assignments though
c868026
@AhmetCanSolak
assigned correct activations in ConvWithBatchNorm implementation
e69b6f3
@AhmetCanSolak
implemented PoolingDown, missing some module assignments though
e93910f
@AhmetCanSolak
upsampling torch call added
e515471
@AhmetCanSolak
handling shiftconv specific additions in the end added
1f29f96
@AhmetCanSolak
SplitAndRot90 implemented
0a89607
@AhmetCanSolak
dropping shiftconv related pieces started
2a98f4d
@AhmetCanSolak
dropping shiftconv related pieces finished
64f3cb2
@AhmetCanSolak
custom_rot implementation deleted as it is not needed anymore with ne…
8ab0547 
…w plan
@AhmetCanSolak
handling skip layers added
e3d24f9
@AhmetCanSolak
assigning right nn implementations in various places
ce6209a
@AhmetCanSolak
added missing tests for torch models
585ca49
@AhmetCanSolak
to keep changes
a25f4c9
@AhmetCanSolak
to keep changes
df40b43
@AhmetCanSolak
calls for convolutions fixed on the second half of the unet
8ef8baf
@AhmetCanSolak
introducing initial test cases for the torch unet model
8c43cae
@AhmetCanSolak
latest changes, test is failing with unmatched tensor shapes on first…
2c7f0b2 
… layer
@AhmetCanSolak
to keep changes
25821cb
@AhmetCanSolak
tests are improved
210f24a
@AhmetCanSolak
lower level method assignment fix on downpooling
79dc3cf
@AhmetCanSolak
fixing right conv module assignment
bee06c3
@AhmetCanSolak
handled the special one time conv on first layer of unet
5715c0b
@AhmetCanSolak AhmetCanSolak changed the title WIP: UNet model in pytorch UNet model in pytorch Feb 4, 2022
@AhmetCanSolak AhmetCanSolak merged commit 9c32e7d into royerlab:master Feb 5, 2022
@AhmetCanSolak AhmetCanSolak deleted the torch-unet-model branch April 12, 2022 18:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@li-li-github li-li-github Awaiting requested review from li-li-github

Assignees

@AhmetCanSolak AhmetCanSolak

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

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