Skip to content

Conversation

@crj1998
Copy link
Contributor

@crj1998 crj1998 commented Dec 8, 2022

Fixes #2794 (comment)

Description:
add unit test for ssim with return type torch.Tensor.

Check list:

  • New tests are added (if a new feature is added)
  • New doc strings: description and/or example code are in RST format
  • Documentation is updated (if required)

@vfdev-5
Copy link
Collaborator

vfdev-5 commented Dec 8, 2022

@crj1998 thanks for the fix ! In addition to that there is a bug in how tests are scheduled, curiously this does not run on CPU but should run.

Copy link
Collaborator

@vfdev-5 vfdev-5 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks @crj1998 !

@vfdev-5 vfdev-5 enabled auto-merge (squash) December 8, 2022 10:28
@vfdev-5 vfdev-5 merged commit edbed90 into pytorch:master Dec 8, 2022
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