vikram71198 commited on
Commit
c4a2161
1 Parent(s): 5614eac

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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