Merge branch 'feature/coref' into coref/dimension-inference

This commit is contained in:
Paul O'Leary McCann 2022-07-12 12:56:10 +09:00 committed by GitHub
commit 64a0bf4460
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -81,7 +81,6 @@ def coref_init(model: Model, X=None, Y=None):
def coref_forward(model: Model, X, is_train: bool):
return model.layers[0](X, is_train)
def convert_coref_clusterer_inputs(model: Model, X: List[Floats2d], is_train: bool):
# The input here is List[Floats2d], one for each doc
# just use the first