MCPcopy Create free account
hub / github.com/pytorch/pytorch / testObserver

Method testObserver

caffe2/python/observer_test.py:30–36  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

28 ws.CreateNet(self.model.net)
29
30 def testObserver(self):
31 ob = self.model.net.AddObserver("TimeObserver")
32 ws.RunNet(self.model.net)
33 print(ob.average_time())
34 num = self.model.net.NumObservers()
35 self.model.net.RemoveObserver(ob)
36 assert(self.model.net.NumObservers() + 1 == num)
37
38 @given(
39 num_layers=st.integers(1, 4),

Callers

nothing calls this directly

Calls 3

AddObserverMethod · 0.80
NumObserversMethod · 0.80
RemoveObserverMethod · 0.80

Tested by

no test coverage detected