adaptdl.goodput module

class adaptdl.goodput.GoodputFunction(perf_params, grad_params, init_batch_size)[source]

Bases: object

efficiency(batch_size)[source]
evaluate(num_nodes, num_replicas, atomic_bsz, accum_steps)[source]
optimize(num_nodes, num_replicas, max_batch_size=None, atomic_bsz_range=None, accumulation=False)[source]
throughput(num_nodes, num_replicas, atomic_bsz, accum_steps)[source]
class adaptdl.goodput.GradParams(sqr, var)

Bases: tuple

property sqr

Alias for field number 0

property var

Alias for field number 1

class adaptdl.goodput.PerfParams(alpha_c, beta_c, alpha_n, beta_n, alpha_r, beta_r, gamma)

Bases: tuple

property alpha_c

Alias for field number 0

property alpha_n

Alias for field number 2

property alpha_r

Alias for field number 4

property beta_c

Alias for field number 1

property beta_n

Alias for field number 3

property beta_r

Alias for field number 5

property gamma

Alias for field number 6

adaptdl.goodput.fit_perf_params(num_nodes, num_replicas, atomic_bsz, accum_step_time, optim_step_time)[source]