in datasets/ClassPrioritySampler.py [0:0]
def get_manual_weights(self, lam):
# Merge balanced weights and uniform weights
if lam == 1:
manual_weights = self.balanced_weights.copy()
elif lam == 0:
manual_weights = self.uniform_weights.copy()
else:
manual_weights = self.balanced_weights * lam + (1-lam) * self.uniform_weights
return manual_weights