GPU Issue textcat.train

spacy

(Madhu Jahagirdar) #1

This is reported in GIt, any workaround in the mean time ?

File “scripts/train_textcat.py”, line 133, in
plac.call(main)
File “/home/motoki/aes/lib/python3.6/site-packages/plac_core.py”, line 328, in call
cmd, result = parser.consume(arglist)
File “/home/motoki/aes/lib/python3.6/site-packages/plac_core.py”, line 207, in consume
return cmd, self.func(*(args + varargs + extraopts), **kwargs)
File “scripts/train_textcat.py”, line 68, in main
losses=losses)
File “/home/motoki/aes/lib/python3.6/site-packages/spacy/language.py”, line 407, in update
proc.update(docs, golds, drop=drop, sgd=get_grads, losses=losses)
File “pipeline.pyx”, line 817, in spacy.pipeline.TextCategorizer.update
File “/home/motoki/aes/lib/python3.6/site-packages/thinc/api.py”, line 61, in begin_update
X, inc_layer_grad = layer.begin_update(X, drop=drop)
File “/home/motoki/aes/lib/python3.6/site-packages/thinc/api.py”, line 176, in begin_update
values = [fwd(X, *a, **k) for fwd in forward]
File “/home/motoki/aes/lib/python3.6/site-packages/thinc/api.py”, line 176, in
values = [fwd(X, *a, **k) for fwd in forward]
File “/home/motoki/aes/lib/python3.6/site-packages/thinc/api.py”, line 258, in wrap
output = func(*args, **kwargs)
File “/home/motoki/aes/lib/python3.6/site-packages/thinc/api.py”, line 61, in begin_update
X, inc_layer_grad = layer.begin_update(X, drop=drop)
File “/home/motoki/aes/lib/python3.6/site-packages/spacy/_ml.py”, line 101, in _preprocess_doc
keys = ops.xp.concatenate(keys)
File “/home/motoki/aes/lib/python3.6/site-packages/cupy/manipulation/join.py”, line 49, in concatenate
return core.concatenate_method(tup, axis)
File “cupy/core/core.pyx”, line 2410, in cupy.core.core.concatenate_method
File “cupy/core/core.pyx”, line 2422, in cupy.core.core.concatenate_method
TypeError: Only cupy arrays can be concatenated


(Matthew Honnibal) #2

No update on that yet sorry – I’ll post to the spaCy tracker when I have a resolution.