FIX: seed rng for test-time dropout

AICONSlab · May 25, 2022 · 4510abd · 4510abd
1 parent c9478bf
commit 4510abd
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/hippmapper/segment/hippmapper.py b/hippmapper/segment/hippmapper.py
@@ -22,9 +22,14 @@
 from nipype.interfaces.fsl import maths
 from nipype.interfaces.c3 import C3d
 from termcolor import colored
+from numpy.random import seed
+from tensorflow import set_random_seed
+
 
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = "3"
 
+seed(1)
+set_random_seed(1)
 
 def parsefn():
     parser = argparse.ArgumentParser(usage="%(prog)s -s [ subj ] \n\n"