inferno-pytorch / inferno

A utility library around PyTorch
Other
244 stars 41 forks source link

Conv 1d and unet refactoring #171

Closed DerThorsten closed 5 years ago

DerThorsten commented 5 years ago

This PR adds the following:

Convolutions:

UNetBase:

This is still a work in progress, the ResUnet still needs the refactoring, also the examples are not yet up-to date

constantinpape commented 5 years ago
* The start block can be used to increase the features from `in_channels` to `initial_features` before the actual unet, the end block is used to convert the output features of the unet to the desired number of channels (@constantinpape what do you think about theses changes)

I like the idea. Didn't have much time to look at the actual changes, will take a closer look on Monday.