nobrainer
nobrainer copied to clipboard
Refactor unet.py to remove hard-coded elements
https://github.com/neuronets/nobrainer/blob/ed0d609333f9aea724c5fd45962c36dfd8a13d88/nobrainer/models/unet.py#L41 and https://github.com/neuronets/nobrainer/blob/ed0d609333f9aea724c5fd45962c36dfd8a13d88/nobrainer/models/unet.py#L27-L39
upon further thinking, it is probably best to leave the hard-coded elements as is because the idea was to present the originally published models as is rather than make them configurable/customizable.
i would leave the defaults to reflect original models but no reason one can't play with some of the architecture.