kd.data.BatchSize

kd.data.BatchSize#

class kauldron.data.BatchSize(total: int)[source]

Bases: object

Batch size.

total

The total batch size across all hosts/devices

Type:

int

per_process

The batch size for a single host

total: int
property per_process: int