Merge pull request #1979 from flairNLP/add-flair-seed

global: add possibility to set flair seed
flairNLP · Nov 23, 2020 · 43b3802 · 43b3802
2 parents 79014b0 + d2c75d1
commit 43b3802
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/flair/__init__.py b/flair/__init__.py
@@ -1,6 +1,7 @@
 import os
 import torch
 from pathlib import Path
+from transformers import set_seed as hf_set_seed
 
 # global variable: cache_root
 cache_root = os.getenv('FLAIR_CACHE_ROOT', Path(Path.home(), ".flair"))
@@ -46,3 +47,6 @@
 )
 
 logger = logging.getLogger("flair")
+
+def set_seed(seed: int):
+    hf_set_seed(seed)