simulation.utils.machine_learning.models.test.test_helper module

Summary

Functions:

main
test_get_norm_layer	Check if get_norm_layer() returns a valid layer.
test_get_scheduler	Check if get_scheduler() returns a scheduler.
test_init_weights	Check if init_weights() runs without errors.
test_set_requires_grad	Check if set_requires_grad() correctly changes requires_grad.

Reference

test_get_norm_layer()

Check if get_norm_layer() returns a valid layer.

test_get_scheduler()

Check if get_scheduler() returns a scheduler.

test_init_weights()

Check if init_weights() runs without errors.

test_set_requires_grad()

Check if set_requires_grad() correctly changes requires_grad.

main()