Skip to content
Snippets Groups Projects
Commit f1b2ea3f authored by Michał Marcińczuk's avatar Michał Marcińczuk
Browse files

Merge branch 'fix-gpu' into 'master'

Fix crash when running infererence on a gpu.

See merge request !18
parents 2a87dbbd e52a1cc8
Branches
Tags v0.4.1
1 merge request!18Fix crash when running infererence on a gpu.
Pipeline #2794 passed with stage
in 4 minutes and 10 seconds
......@@ -92,7 +92,7 @@ class PolDeepNer2:
# state_dict = torch.load(open(os.path.join(model_path, 'model.pt'), 'rb'))
model.load_state_dict(state_dict)
#model.eval()
#model.to(device)
model.to(device)
self.model = model
self.device = device
self.squeeze = squeeze
......
......@@ -19,7 +19,7 @@ install_requires = [
setuptools.setup(
name="poldeepner2",
version="0.4.0",
version="0.4.1",
author="Michał Marcińczuk",
author_email="michal.marcinczuk@pwr.edu.pl",
description="PolDeepNer2 is a tool for sequence labeling tasks based on transformer language models.",
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment