search

adambielski / siamese-triplet

Siamese and triplet networks with online pair/triplet mining in PyTorch
BSD 3-Clause "New" or "Revised" License
3.1k stars 633 forks source link

BalancedBatchSampler: classes vs samples #65

Open sk-zk opened 3 years ago

sk-zk commented 3 years ago

I'm extremely new to machine learning so this might be a bit of a dumb question, but I'm curious anyway:

I don't have a lot of GPU memory. For 224x224 images, should I increase n_classes and lower n_samples, or the other way around?

milky-tea-1997 commented 2 years ago

Hi, have you solved the problem yet? In fact, I have the same difficulty