From 2d3c559dc4bb4bf5d829a748634afdef79e4bd0f Mon Sep 17 00:00:00 2001
From: Paul O'Leary McCann <polm@dampfkraft.com>
Date: Sat, 3 Jul 2021 18:43:03 +0900
Subject: [PATCH] On initialize, use just two samples

Coref docs are kind of long, and using 10 samples on a smallish GPU can
cause OOMs.
---
 spacy/pipeline/coref.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spacy/pipeline/coref.py b/spacy/pipeline/coref.py
index f0ae62fa9..2f9baaeb4 100644
--- a/spacy/pipeline/coref.py
+++ b/spacy/pipeline/coref.py
@@ -333,7 +333,7 @@ class CoreferenceResolver(TrainablePipe):
 
         X = []
         Y = []
-        for ex in islice(get_examples(), 10):
+        for ex in islice(get_examples(), 2):
             X.append(ex.predicted)
             Y.append(ex.reference)