Have you checked if the weights are not changing? A crude way to do this is to sum all of the parameters of the model as a “checksum.”
def checksum(model):
    s = 0.0
    for param in model.parameters():
        s += torch.sum(param)
    return s