We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 84615ea commit 62d5609Copy full SHA for 62d5609
paddlenlp/trainer/plugins/timer.py
@@ -32,15 +32,15 @@ def __init__(self, name):
32
def start(self):
33
"""Start the timer."""
34
assert not self.started_, f"{self.name} timer has already started"
35
- if "gpu" in paddle.device.get_device():
+ if "cpu" not in paddle.device.get_device():
36
paddle.device.synchronize()
37
self.start_time = time.time()
38
self.started_ = True
39
40
def stop(self):
41
"""Stop the timers."""
42
assert self.started_, f"{self.name} timer is not started."
43
44
45
self.elapsed_ += time.time() - self.start_time
46
self.started_ = False
0 commit comments