(self)
| 153 | - net2.params[name][i].data).sum(), 0) |
| 154 | |
| 155 | def test_save_hdf5(self): |
| 156 | f = tempfile.NamedTemporaryFile(mode='w+', delete=False) |
| 157 | f.close() |
| 158 | self.net.save_hdf5(f.name) |
| 159 | net_file = simple_net_file(self.num_output) |
| 160 | net2 = caffe.Net(net_file, caffe.TRAIN) |
| 161 | net2.load_hdf5(f.name) |
| 162 | os.remove(net_file) |
| 163 | os.remove(f.name) |
| 164 | for name in self.net.params: |
| 165 | for i in range(len(self.net.params[name])): |
| 166 | self.assertEqual(abs(self.net.params[name][i].data |
| 167 | - net2.params[name][i].data).sum(), 0) |
| 168 | |
| 169 | class TestLevels(unittest.TestCase): |
| 170 |
nothing calls this directly
no test coverage detected