Skip to content

Commit ead7476

Browse files
modify argument palce to device
1 parent 8aea4e1 commit ead7476

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/paddle/fluid/dygraph/layers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1370,7 +1370,7 @@ def _apply(self, func, device, dtype, blocking):
13701370

13711371
def to(self, device=None, dtype=None, blocking=None):
13721372
'''
1373-
Cast the parameters and buffers of Layer by the give place, dtype and blocking.
1373+
Cast the parameters and buffers of Layer by the give device, dtype and blocking.
13741374
13751375
Parameters:
13761376
device(str|paddle.CPUPlace()|paddle.CUDAPlace()|paddle.CUDAPinnedPlace()|None, optional): The device of the Layer which want to be stored.

0 commit comments

Comments
 (0)