cerebras.modelzoo.common.utils.run.utils.DeviceType#

class cerebras.modelzoo.common.utils.run.utils.DeviceType[source]#

Bases: object

Supported Devices for Running Modelzoo Scripts

Methods

devices

Valid strategies

Attributes

ANY

CPU

CSX

GPU

classmethod devices()[source]#

Valid strategies

previous

cerebras.modelzoo.common.utils.run.utils.update_runconfig_debug_args_path

next

cerebras.modelzoo.common.utils.run.write_weight_summaries