Skip to content

Conversation

@pmeier
Copy link
Contributor

@pmeier pmeier commented Aug 22, 2022

I dropped the ball for two separate things in #6447:

  • PIL is not supported by MixUp and CutMix since we need a batch of images (read batched and collated).
  • Just adding plain tensors to the allowed types is not sufficient and _transforms has to be adapted.

Copy link
Contributor

@datumbox datumbox left a comment

Choose a reason for hiding this comment

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

LGTM, sorry I didn't catch it too on the review.

@pmeier pmeier merged commit 6279089 into pytorch:main Aug 23, 2022
@pmeier pmeier deleted the cutmix-mixup-pil branch August 23, 2022 07:06
@pmeier pmeier mentioned this pull request Aug 23, 2022
facebook-github-bot pushed a commit that referenced this pull request Aug 25, 2022
Summary:
* fix MixUp and CutMix

* improve error message

Reviewed By: datumbox

Differential Revision: D39013676

fbshipit-source-id: 648fea88080ce81d98e3355fe61a7f195121d582
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants