|
|
|
|
|
|
|
|
|
|
| import numpy as np |
|
|
| import caffe2.python.models.resnet as resnet |
| import hypothesis.strategies as st |
| from hypothesis import given, settings |
| import caffe2.python.hypothesis_test_util as hu |
| import caffe2.python.models.imagenet_trainer_test_utils as utils |
|
|
| class ResnetMemongerTest(hu.HypothesisTestCase): |
|
|
| @given(with_shapes=st.booleans(), **hu.gcs_cpu_only) |
| @settings(max_examples=2, deadline=None) |
| def test_resnet_shared_grads(self, with_shapes, gc, dc): |
| results = utils.test_shared_grads( |
| with_shapes, |
| resnet.create_resnet50, |
| 'gpu_0/conv1_w', |
| 'gpu_0/last_out_L1000' |
| ) |
| self.assertTrue(results[0][0] < results[0][1]) |
| np.testing.assert_almost_equal(results[1][0], results[1][1]) |
| np.testing.assert_almost_equal(results[2][0], results[2][1]) |
|
|
| def test_resnet_forward_only(self): |
| results = utils.test_forward_only( |
| resnet.create_resnet50, |
| 'gpu_0/last_out_L1000' |
| ) |
| self.assertTrue(results[0][0] < results[0][1]) |
| self.assertTrue(results[1] < 7 and results[1] > 0) |
| np.testing.assert_almost_equal(results[2][0], results[2][1]) |
|
|
| def test_resnet_forward_only_fast_simplenet(self): |
| ''' |
| Test C++ memonger that is only for simple nets |
| ''' |
| results = utils.test_forward_only_fast_simplenet( |
| resnet.create_resnet50, |
| 'gpu_0/last_out_L1000' |
| ) |
|
|
| self.assertTrue(results[0][0] < results[0][1]) |
| self.assertTrue(results[1] < 4 and results[1] > 0) |
| np.testing.assert_almost_equal(results[2][0], results[2][1]) |
|
|
|
|
| if __name__ == "__main__": |
| import unittest |
| import random |
| random.seed(2603) |
| |
| workspace.GlobalInit([ |
| 'caffe2', |
| '--caffe2_log_level=0', |
| '--caffe2_print_blob_sizes_at_exit=0', |
| '--caffe2_gpu_memory_tracking=1']) |
| unittest.main() |
|
|
