Skip to content

Commit eb6c60c

Browse files
authored
fix(nyz): fix default max step dtype bug (#163)
1 parent 954d310 commit eb6c60c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ding/framework/task.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def use_step_wrapper(self, fn: Callable) -> 'Task':
118118
self.step_wrappers.append(fn)
119119
return self
120120

121-
def run(self, max_step: int = 1e10) -> None:
121+
def run(self, max_step: int = int(1e10)) -> None:
122122
"""
123123
Overview:
124124
Execute the iterations, when reach the max_step or task.finish is true,

0 commit comments

Comments
 (0)