|
|
|
|
|
|
|
|
|
|
| import unittest |
| from hypothesis import given |
| import hypothesis.strategies as st |
| import numpy as np |
| from caffe2.python import core, workspace |
| import caffe2.python.hypothesis_test_util as hu |
| import caffe2.python.ideep_test_util as mu |
|
|
|
|
| @unittest.skipIf(not workspace.C.use_mkldnn, "No MKLDNN support.") |
| class DropoutTest(hu.HypothesisTestCase): |
|
|
| @given(X=hu.tensor(), |
| in_place=st.booleans(), |
| ratio=st.floats(0, 0.999), |
| **mu.gcs) |
| def test_dropout_is_test(self, X, in_place, ratio, gc, dc): |
| """Test with is_test=True for a deterministic reference impl.""" |
| op = core.CreateOperator('Dropout', ['X'], |
| ['X' if in_place else 'Y'], |
| ratio=ratio, is_test=True) |
|
|
| self.assertDeviceChecks(dc, op, [X], [0]) |
| |
|
|
| def reference_dropout_test(x): |
| return x, np.ones(x.shape, dtype=np.bool) |
| self.assertReferenceChecks( |
| gc, op, [X], reference_dropout_test, |
| |
| outputs_to_check=[0]) |
|
|
| @given(X=hu.tensor(), |
| in_place=st.booleans(), |
| output_mask=st.booleans(), |
| **mu.gcs) |
| @unittest.skipIf(True, "Skip duo to different rand seed.") |
| def test_dropout_ratio0(self, X, in_place, output_mask, gc, dc): |
| """Test with ratio=0 for a deterministic reference impl.""" |
| is_test = not output_mask |
| op = core.CreateOperator('Dropout', ['X'], |
| ['X' if in_place else 'Y'] + |
| (['mask'] if output_mask else []), |
| ratio=0.0, is_test=is_test) |
|
|
| self.assertDeviceChecks(dc, op, [X], [0]) |
|
|
| def reference_dropout_ratio0(x): |
| return (x,) if is_test else (x, np.ones(x.shape, dtype=np.bool)) |
| self.assertReferenceChecks( |
| gc, op, [X], reference_dropout_ratio0, outputs_to_check=[0]) |
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
