modelzoo.common.pytorch.input_utils

Functions

bucketed_batch	Batch the data from an iterator such that sampels of similar length end up in the same batch.
get_streaming_batch_size	Returns the streaming batch size of the current task.
validate_streaming_and_micro_batch_size	Validates the streaming and micro batch sizes.

previous

modelzoo.common.pytorch.half_dtype.HalfDType

next

modelzoo.common.pytorch.input_utils.bucketed_batch