(self)
| 848 | self.assertEqual(predictions[0].shape, desired_shape) |
| 849 | |
| 850 | def test_set_dropout_mode(self): |
| 851 | # set dropmout mode for all dropout layers |
| 852 | for train_mode in [True, False]: |
| 853 | model = Torch_Model() |
| 854 | modules = list(model.modules()) |
| 855 | |
| 856 | for module in modules: |
| 857 | self.assertEqual(module.training, True) |
| 858 | |
| 859 | modAL.dropout.set_dropout_mode(model, [], train_mode) |
| 860 | |
| 861 | self.assertEqual(modules[7].training, train_mode) |
| 862 | self.assertEqual(modules[11].training, train_mode) |
| 863 | |
| 864 | # set dropout mode only for special layers: |
| 865 | for train_mode in [True, False]: |
| 866 | model = Torch_Model() |
| 867 | modules = list(model.modules()) |
| 868 | modAL.dropout.set_dropout_mode(model, [7], train_mode) |
| 869 | self.assertEqual(modules[7].training, train_mode) |
| 870 | self.assertEqual(modules[11].training, True) |
| 871 | |
| 872 | modAL.dropout.set_dropout_mode(model, [], True) |
| 873 | modAL.dropout.set_dropout_mode(model, [11], train_mode) |
| 874 | self.assertEqual(modules[11].training, train_mode) |
| 875 | self.assertEqual(modules[7].training, True) |
| 876 | |
| 877 | # No Dropout Layer |
| 878 | self.assertRaises(KeyError, modAL.dropout.set_dropout_mode, |
| 879 | model, [5], train_mode) |
| 880 | |
| 881 | |
| 882 | class TestDeepActiveLearner(unittest.TestCase): |
nothing calls this directly
no test coverage detected