wba.rich:

shuffle = True


apart from anything else, you are using shuffle in the dataloader parameters which will randomize the inputs sequence. So if you test it for a little training time, you will likely get different results from first example.