cerebras.modelzoo.common.utils.utils.configure_selective_gradient

cerebras.modelzoo.common.utils.utils.configure_selective_gradient(config)

Takes in a dictionary of selective grad configs and returns the output as a list of SelectiveGrad

previous

cerebras.modelzoo.common.utils.utils.configure_compression

next

cerebras.modelzoo.common.utils.utils.format_rate