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.

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

Refactor unet.py to remove hard-coded elements #280

Open
hvgazula opened this issue Mar 5, 2024 · 2 comments
Open

Refactor unet.py to remove hard-coded elements #280

hvgazula opened this issue Mar 5, 2024 · 2 comments
Assignees

Comments

@hvgazula
Copy link
Contributor

hvgazula commented Mar 5, 2024

n_base_filters = 16

and
conv_kwds = {
"kernel_size": (3, 3, 3),
"activation": None,
"padding": "same",
# 'kernel_regularizer': tf.keras.regularizers.l2(0.001),
}
conv_transpose_kwds = {
"kernel_size": (2, 2, 2),
"strides": 2,
"padding": "same",
# 'kernel_regularizer': tf.keras.regularizers.l2(0.001),
}

@hvgazula hvgazula added the bug label Mar 5, 2024
@hvgazula hvgazula self-assigned this Mar 5, 2024
@hvgazula
Copy link
Contributor Author

hvgazula commented Mar 5, 2024

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.

@hvgazula hvgazula added enhancement and removed bug labels Mar 5, 2024
@satra
Copy link
Contributor

satra commented Mar 6, 2024

i would leave the defaults to reflect original models but no reason one can't play with some of the architecture.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants