Seed for neural nets

Alphathon
#1

Hi, Im trying to make some tensorflow model learning reproducible but I got the following error when trying :

from tensorflow import set_random_seed
set_random_seed(2)

ImportError: cannot import name ‘set_random_seed’

Maybe this is not the right way to seed the model.
Any help appreciated :slight_smile: thanks

#2

Hi narcejac,

Could you please try:

from tensorflow.random import set_seed
set_seed(2)

Please refer to this stack overflow for more information.

Thanks!