explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-18 20:22:25 +03:00
Code Issues Packages Projects Releases Wiki Activity

Import torch from thinc

Browse Source
This commit is contained in:
Paul O'Leary McCann 2022-05-24 17:28:27 +09:00
parent 5cbc9f4573
commit c9233a5a1f
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/ml/models/coref.py
View File

@ -1,10 +1,9 @@
from typing import List, Tuple
import torch
from thinc.api import Model, chain
from thinc.api import PyTorchWrapper, ArgsKwargs
from thinc.types import Floats2d, Ints2d, Ints1d
from thinc.util import xp2torch, torch2xp
from thinc.util import torch, xp2torch, torch2xp
from ...tokens import Doc
from ...util import registry

3
spacy/ml/models/span_predictor.py
View File

@ -1,10 +1,9 @@
from typing import List, Tuple
import torch
from thinc.api import Model, chain, tuplify
from thinc.api import PyTorchWrapper, ArgsKwargs
from thinc.types import Floats2d, Ints1d
from thinc.util import xp2torch, torch2xp
from thinc.util import torch, xp2torch, torch2xp
from ...tokens import Doc
from ...util import registry