Skip to content

Conversation

@adam9500370
Copy link
Contributor

  • Directly interpolate input tensor when input size and target size are different in cross_entropy2d
  • Check device type by target tensor in multi_scale_cross_entropy2d
  • Replace with_bn with is_batchnorm for name consistency

@meetps meetps merged commit 1ce2cc0 into meetps:master Nov 23, 2018
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.

2 participants