Skip to content

refactor how masks are retrieved#1066

Merged
bghira merged 2 commits intomainfrom
refactor/masking-loss-input-retrieval
Oct 13, 2024
Merged

refactor how masks are retrieved#1066
bghira merged 2 commits intomainfrom
refactor/masking-loss-input-retrieval

Conversation

@bghira
Copy link
Owner

@bghira bghira commented Oct 13, 2024

when square cropping, it worked alright, but when aspect bucketing with random crops or random/closest bucket, the mask would not prepare the same as the origin sample.

a prepare_like method is added to prepare one sample using the same transforms as another.

optimises startup speed of larger datasets by skipping the aspect bucketing and VAE caching steps which are not necessary.

@bghira bghira merged commit a81d449 into main Oct 13, 2024
@bghira bghira deleted the refactor/masking-loss-input-retrieval branch October 14, 2024 01:49
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