SkyWater21 commited on
Commit
3db324f
1 Parent(s): 5c3af3a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -0
README.md CHANGED
@@ -42,6 +42,16 @@ Labels:
42
  27: neutral
43
  ```
44
 
 
 
 
 
 
 
 
 
 
 
45
  Training parameters:
46
  ```yaml
47
  max_length: null
 
42
  27: neutral
43
  ```
44
 
45
+ Seed used for random number generator is 42:
46
+ ```python
47
+ def set_seed(seed=42):
48
+ random.seed(seed)
49
+ np.random.seed(seed)
50
+ torch.manual_seed(seed)
51
+ if torch.cuda.is_available():
52
+ torch.cuda.manual_seed_all(seed)
53
+ ```
54
+
55
  Training parameters:
56
  ```yaml
57
  max_length: null