This adds a unittest designed to test basic train/save/load functionality of fnet_model. Functionality is tested on single GPUs, multiple GPUs, and on the CPU. Also, the test checks that a small (4-parameter) saved model (added to the repository) is loadable, which is designed to represent "legacy" models in case save/load functionality is changed.
A new script was added (tests/run_tests.sh) for running all tests from the root project directory.
This adds a unittest designed to test basic train/save/load functionality of
fnet_model
. Functionality is tested on single GPUs, multiple GPUs, and on the CPU. Also, the test checks that a small (4-parameter) saved model (added to the repository) is loadable, which is designed to represent "legacy" models in case save/load functionality is changed.A new script was added (
tests/run_tests.sh
) for running all tests from the root project directory.