cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms.RenameTransform#

class cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms.RenameTransform(in_key, out_key, delete_old=False)[source]#

Bases: object

Saves the value of data_dict[in_key] to data_dict[out_key]. Optionally removes data_dict[in_key] from the dict.

Methods