This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
Register
Sign In
hummingbird
/
fastNLP
Not watched
Unwatch
Watch all
Watch but not notify
2
Star
0
Fork
0
Code
Releases
13
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
Browse Source
add nll loss
tags/v0.1.0
2017alan
6 years ago
parent
82502aa67d
commit
8c13c28f0c
1 changed files
with
2 additions
and
0 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+2
-0
fastNLP/core/loss.py
+ 2
- 0
fastNLP/core/loss.py
View File
@@ -37,5 +37,7 @@ class Loss(object):
"""
if loss_name == "cross_entropy":
return torch.nn.CrossEntropyLoss()
elif loss_name == 'nll':
return torch.nn.NLLLoss()
else:
raise NotImplementedError
Write
Preview
Loading…
Cancel
Save