gsnn.gsnn.tests.test_diagnostics

Tests for training diagnostics utilities.

Functions

test_grad_diagnostics_analyze(minimal_gsnn)
test_grad_diagnostics_update_reset(minimal_gsnn)
test_training_diagnostics_smoke(minimal_gsnn)

Classes

GSNN(*args, **kwargs)
GradDiagnostics([window_size, verbose])	Gradient Diagnostics for Vanishing Gradient Analysis in PyTorch Models.

gsnn.gsnn.tests.test_diagnostics.test_grad_diagnostics_analyze(minimal_gsnn)

gsnn.gsnn.tests.test_diagnostics.test_grad_diagnostics_update_reset(minimal_gsnn)

gsnn.gsnn.tests.test_diagnostics.test_training_diagnostics_smoke(minimal_gsnn)