-
Notifications
You must be signed in to change notification settings - Fork 1
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
Upsample mode parameter for UpBlock #76
Conversation
Test Results 1 files 1 suites 22m 51s ⏱️ Results for commit 32824d0. ♻️ This comment has been updated with latest results. |
☂️ Python Coverage
Overall Coverage
New FilesNo new covered files... Modified Files
|
luxonis_train/nodes/blocks/blocks.py
Outdated
ConvModule(out_channels, out_channels, kernel_size=3, padding=1), | ||
layers = [] | ||
|
||
if upsample_mode == "transpose": |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should this be "cov_transpose"
?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Oh, right, thanks!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
Co-authored-by: GitHub Actions <actions@github.com>
This PR adds an
upsample_mode
parameter to theUpBlock
, which is used in the default segmentation head, to avoid checkerboard artifacts caused by thenn.ConvTranspose2d
layer.Now,
nn.Upsample
is the default choice in theUpBlock
.