浏览代码

Add seeding for numpy functions

dustin.born 6 年之前
父节点
当前提交
25bc234c5c
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      code/Attack/BaseAttack.py

+ 1 - 0
code/Attack/BaseAttack.py

@@ -292,6 +292,7 @@ class BaseAttack(metaclass=abc.ABCMeta):
         """
         if isinstance(seed, int):
             random.seed(seed)
+            np.random.seed(seed)
 
     def set_start_time(self):
         """