cerebras.modelzoo.common.pytorch_utils

General purpose Pytorch Utilities.

Functions

to_cpu	Move tensor from device to cpu.
to_tensor	If the provided value is a Python int or float, it converts them into PyTorch Tensors of type int32 and float32 respectively.
visit_structure	Recursively traverse nested structure and return the items accepted by the selector.

Classes

BufferedShuffleDataset	Dataset shuffled from the original dataset.
IterableDatasetSampler	This sampler can be used with a multi-worker distributed dataloader.
RunConfigParamsValidator	Validate Run Configs.
SampleGenerator	Iterator which returns multiple samples of a given input data.

previous

cerebras.modelzoo.common.model_utils.count_lines.count_lines

next

cerebras.modelzoo.common.pytorch_utils.to_cpu