in smdebug_rulesconfig/profiler_rules/rules.py [0:0]
def get_rules(cls):
return [
BatchSize,
CPUBottleneck,
Dataloader,
GPUMemoryIncrease,
IOBottleneck,
LoadBalancing,
LowGPUUtilization,
MaxInitializationTime,
OverallSystemUsage,
StepOutlier,
]