MCPcopy Index your code
hub / github.com/pytorch/pytorch / test_dropout

Method test_dropout

caffe2/python/brew_test.py:33–42  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

31 self.model = ModelHelper(name="test_model")
32
33 def test_dropout(self):
34 p = 0.2
35 X = np.ones((100, 100)).astype(np.float32) - p
36 workspace.FeedBlob("x", X)
37 model = ModelHelper(name="test_model")
38 brew.dropout(model, "x", "out", is_test=False)
39 workspace.RunNetOnce(model.param_init_net)
40 workspace.RunNetOnce(model.net)
41 out = workspace.FetchBlob("out")
42 self.assertLess(abs(out.mean() - (1 - p)), 0.05)
43
44 def test_fc(self):
45 m, n, k = (15, 15, 15)

Callers

nothing calls this directly

Calls 5

ModelHelperClass · 0.90
astypeMethod · 0.80
absFunction · 0.50
onesMethod · 0.45
meanMethod · 0.45

Tested by

no test coverage detected