Update README.md
Browse files
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
|