Skip to content

Fix the kernel size of head, add align_corners=False for interpolate#40

Open
D-X-Y wants to merge 1 commit into
HRNet:masterfrom
D-X-Y:master
Open

Fix the kernel size of head, add align_corners=False for interpolate#40
D-X-Y wants to merge 1 commit into
HRNet:masterfrom
D-X-Y:master

Conversation

@D-X-Y

@D-X-Y D-X-Y commented Dec 3, 2019

Copy link
Copy Markdown

Change two lines:

L241 in lib/models/hrnet.py, explicitly use align_corners=False, otherwise, it will raise a warning in PyTorch>1.3.0

L309, the kernel_size should be extra.FINAL_CONV_KERNEL (to align with L311).

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant