@Sebastian_E I had the same issue back then, with the exact same seeding function that you used. There might be two problems:


To debug, you can just set to one epoch and evaluate first. Once you get the predicted value, record it down. Then proceed to restart the notebook and train once again (provided your data is very small and 1 epoch takes less than a few minutes). Compare the answer.


Colab has different GPUs, every GPU might yield different results even if you set the seed. Try to ensure you are using the same GPU on colab by:



gpu_info = !nvidia-smi
gpu_info = '\n'.join(gpu_info)
if gpu_info.find('failed') >= 0:
    print('Select the Runtime > "Change runtime type" menu to enable a GPU accelerator, ')
    print('and then re-execute this cell.')
else:
    print(gpu_info)