I’m using the DataLoader, so I don’t really have all of the samples loaded at once - only batch per batch. And those batches are randomly loaded, so isn’t it the same to have my pos_weight be per batch and per dataset?