vikram71198
commited on
Commit
•
c4a2161
1
Parent(s):
5614eac
Update README.md
Browse files
README.md
CHANGED
@@ -69,7 +69,7 @@ for x,y in prediction_probabilities:
|
|
69 |
predictions.append("not_fake_news") if x > y else predictions.append("fake_news")
|
70 |
print(predictions)
|
71 |
```
|
72 |
-
|
73 |
### Framework versions
|
74 |
|
75 |
- Transformers 4.12.5
|
|
|
69 |
predictions.append("not_fake_news") if x > y else predictions.append("fake_news")
|
70 |
print(predictions)
|
71 |
```
|
72 |
+
Please note that if you're performing inference on a lengthy dataset, split it up into multiple batches, otherwise your RAM will overflow, unless you're using a really high end GPU/TPU setup. I'd recommend a batch length of 50, if you're working with a vanilla GPU setup.
|
73 |
### Framework versions
|
74 |
|
75 |
- Transformers 4.12.5
|