(self)
| 48 | return (model, perfect_model, data, label) |
| 49 | |
| 50 | def testDense(self): |
| 51 | model, perfect_model, data, label = self._createDense() |
| 52 | optimizer = self.build_optimizer(model) |
| 53 | workspace.FeedBlob('data', data[0]) |
| 54 | workspace.FeedBlob('label', label[0]) |
| 55 | workspace.RunNetOnce(model.param_init_net) |
| 56 | workspace.CreateNet(model.net, True) |
| 57 | for _ in range(2000): |
| 58 | idx = np.random.randint(data.shape[0]) |
| 59 | workspace.FeedBlob('data', data[idx]) |
| 60 | workspace.FeedBlob('label', label[idx]) |
| 61 | workspace.RunNet(model.net.Proto().name) |
| 62 | |
| 63 | np.testing.assert_allclose( |
| 64 | perfect_model[np.newaxis, :], |
| 65 | workspace.FetchBlob('fc_w'), |
| 66 | atol=1e-2 |
| 67 | ) |
| 68 | self.check_optimizer(optimizer) |
| 69 | |
| 70 | @unittest.skipIf(not workspace.has_gpu_support, "No gpu support") |
| 71 | def testGPUDense(self, dtype=core.DataType.FLOAT): |
nothing calls this directly
no test coverage detected