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
删除PaddleDriver.zero_grad对set_to_none的warning
tags/v1.0.0alpha
x54-729
2 years ago
parent
f943f17e60
commit
484b932772
1 changed files
with
0 additions
and
2 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+0
-2
fastNLP/core/drivers/paddle_driver/paddle_driver.py
+ 0
- 2
fastNLP/core/drivers/paddle_driver/paddle_driver.py
View File
@@ -71,8 +71,6 @@ class PaddleDriver(Driver):
:param set_to_none: 用来判断是否需要将梯度直接置为 None;Paddle中这个参数无效。
"""
if set_to_none:
logger.rank_zero_warning("Parameter `set_to_none` does nothing in paddle since grad cannot be set directly.")
for optimizer in self.optimizers:
optimizer.clear_grad()
Write
Preview
Loading…
Cancel
Save