소스 검색

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):
         """