balbasty / nitorch

Neuroimaging in PyTorch
Other
86 stars 14 forks source link

Refactor: nn.segment #55

Closed balbasty closed 3 years ago

balbasty commented 3 years ago

Another large PR.

This one includes a pretty big refactoring of nn, in particular the generators submodule, as well as modules.conv and modules.segmentation.

There will very probably be breaking changes (at least for imports). I'll wait for one or two weeks before merging this one.