======================== short test summary info ========================
FAILED tests/modules/test_layers.py::TestModuleLayers::test_bert_classifier_head
FAILED tests/modules/test_layers.py::TestModuleLayers::test_mlp - Asse...
FAILED tests/utils/test_checkpoint.py::TestUtilsCheckpoint::test_save_and_load_state_dict
For test_checkpoint.py: E RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'other' in call to _th_equal
For other cases, I see errors like this:
> output.squeeze().tolist(), [0.5452202, -0.0437842, -0.377468], decimal=3
)
E AssertionError:
E Arrays are not almost equal to 3 decimals
E
E Mismatched elements: 3 / 3 (100%)
E Max absolute difference: 0.45450824
E Max relative difference: 10.38064516
E x: array([ 0.741, 0.411, -0.254])
E y: array([ 0.545, -0.044, -0.377])
tests/modules/test_layers.py:114: AssertionError
When I clone the master and run as per instructions here: https://mmf.readthedocs.io/en/latest/notes/installation.html
I see following test failures with pytest:
For test_checkpoint.py: E RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'other' in call to _th_equal
For other cases, I see errors like this: