confused with the code and ability of the weight_learner fuction in reweighting.py

[pending1face]

I print the softmax_weight like below:

and the result is here

Does this mean that it did not have the ability to reweight the loss of each sample？

Also，I print the loss in weight calculating process like below:

The results show that this part does not seem to learn well:

[universe27king]

I also found that the weights of each samples appear almost identical, do you know why is this now? thank you.