mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 13:44:15 +08:00
81b9aa743b
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52083 This makes minor fixes in `caffe2/python` to address all errors currently reported by Pyre. I update the code to fix errors when doing so looked simple and safe, and added `pyre-fixme` comments in other places. ghstack-source-id: 121109695 Test Plan: Confirmed that Pyre no longer reports errors under `caffe2/python` Differential Revision: D26272279 fbshipit-source-id: b1eb19d323b613f23280ce9c71e800e874ca1162
63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
|
|
|
|
|
|
|
|
|
|
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)
|
|
# pyre-fixme[10]: Name `workspace` is used but not defined in the current scope
|
|
workspace.GlobalInit([
|
|
'caffe2',
|
|
'--caffe2_log_level=0',
|
|
'--caffe2_print_blob_sizes_at_exit=0',
|
|
'--caffe2_gpu_memory_tracking=1'])
|
|
unittest.main()
|