Solved by tlwill11 in post #2 
              
              
                Nevermind. I wrote a small def to insert the weights I wanted (or think I want) per batch. The code is here: 
def weight_(x):
    x = x.cpu()
    wn = len(train_indices)/n_neg
    wp = len(train_indices)/n_pos
    x = np.where(x == 0, wn, x)
    x = np.where(x == 1, wp, x)
    x = torch.from_numpy(x…