Model not training: Model predicting same thing every time

[J_Johnson]

I’ve added a reply to the FAQ which gives a function for calculating the rescaling weight vector used by the CrossEntropyLoss function in PyTorch.

You can find that here:

How do I get the accuracy of an unbalanced dataset or segmentation task?

Here is a function for calculating the weight vector used by CrossEntropyLoss based on a set of targets with integers. def scaled_weights(targets): ''' Gets class weights :param targets: pass in the matrix of size (batch, classes) containing correct targets or labels as one_hot :return: returns a vector for weight rescaling for use in cross entropy loss function ''' N, C = targets.size(0), targets.size(1) # get vector of total for each class totals = ta…