Skip to content

Commit 511b9f8

Browse files
committed
floor mult for max_calls_split
1 parent 1ded892 commit 511b9f8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

library/train_util.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -4881,7 +4881,7 @@ def schedule_timesteps(max_calls, min_timestep, max_timestep, restarts=0, b_size
48814881
if not hasattr(schedule_timesteps, 'current_step'):
48824882
schedule_timesteps.current_step = 1
48834883

4884-
max_calls_split = max_calls / (restarts + 1)
4884+
max_calls_split = max_calls // (restarts + 1)
48854885
interpolation_factor = schedule_timesteps.current_step / max_calls_split
48864886

48874887
# Calculate mode for the triangular distribution

0 commit comments

Comments
 (0)