cerebras.modelzoo.config_manager.config_classes.base.run_config.CSTorchProfilerConfig#

class cerebras.modelzoo.config_manager.config_classes.base.run_config.CSTorchProfilerConfig(host_activities: Optional[List[str]], start_step: int = <object object at 0x7f0436677b60>, end_step: int = <object object at 0x7f0436677b60>)[source]#
host_activities: Optional[List[str]]#

list of act host, wgt host, csx host in the form of strings which should like, ACTHOST0, WGTHOST0, CSX0, etc

start_step: int = <object object>#

step where to begin profiling

end_step: int = <object object>#

step where to end profiling