Some metrics are better to have separate by high-level modules: - grad norm
Some metrics are better to have separate by high-level modules: