Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
poldeepner2
Manage
Activity
Members
Labels
Plan
Issues
29
Issue boards
Milestones
Wiki
Redmine
Code
Merge requests
0
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Container Registry
Operate
Environments
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
Information extraction
poldeepner2
Commits
f1b2ea3f
Commit
f1b2ea3f
authored
4 years ago
by
Michał Marcińczuk
Browse files
Options
Downloads
Plain Diff
Merge branch 'fix-gpu' into 'master'
Fix crash when running infererence on a gpu. See merge request
!18
parents
2a87dbbd
e52a1cc8
Branches
Branches containing commit
Tags
v0.4.1
Tags containing commit
1 merge request
!18
Fix crash when running infererence on a gpu.
Pipeline
#2794
passed with stage
Stage:
in 4 minutes and 10 seconds
Changes
2
Pipelines
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
poldeepner2/models.py
+1
-1
1 addition, 1 deletion
poldeepner2/models.py
setup.py
+1
-1
1 addition, 1 deletion
setup.py
with
2 additions
and
2 deletions
poldeepner2/models.py
+
1
−
1
View file @
f1b2ea3f
...
...
@@ -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
...
...
This diff is collapsed.
Click to expand it.
setup.py
+
1
−
1
View file @
f1b2ea3f
...
...
@@ -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.
"
,
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment