Update README.md
Browse files
README.md
CHANGED
@@ -86,7 +86,10 @@ label_dict, label_n2w = read_labelFile(labelFile)
|
|
86 |
|
87 |
## Trainer
|
88 |
```python
|
89 |
-
|
|
|
|
|
|
|
90 |
model = my_model()
|
91 |
model.to(device)
|
92 |
print(model)
|
|
|
86 |
|
87 |
## Trainer
|
88 |
```python
|
89 |
+
# set the seed for ensuring reproducibility
|
90 |
+
seed = 3407
|
91 |
+
# init net
|
92 |
+
print('init net...')
|
93 |
model = my_model()
|
94 |
model.to(device)
|
95 |
print(model)
|