modelzoo.common.pytorch.run_cstorch_flow.compute_grad_norm#

modelzoo.common.pytorch.run_cstorch_flow.compute_grad_norm(model)[source]#

Compute the model wise and per layer norm of the gradients

previous

modelzoo.common.pytorch.run_cstorch_flow

next

modelzoo.common.pytorch.run_cstorch_flow.compute_params_norm