Skip to content
This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

train models with multicrops #122

Open
lxysl opened this issue Oct 13, 2023 · 0 comments
Open

train models with multicrops #122

lxysl opened this issue Oct 13, 2023 · 0 comments

Comments

@lxysl
Copy link

lxysl commented Oct 13, 2023

I wonder why can directly inference the model with a list? In main_swav.py line 284

for it, inputs in enumerate(train_loader):
    embedding, output = model(inputs)

Doesn't the multicrop dataloader yield a list of tensors with different sized image crop?

    def __getitem__(self, index):
        path, _ = self.samples[index]
        image = self.loader(path)
        multi_crops = list(map(lambda trans: trans(image), self.trans))
        if self.return_index:
            return index, multi_crops
        return multi_crops
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant