(self)
| 136 | ('loss', ['ip_blob', 'label'])])) |
| 137 | |
| 138 | def test_save_and_read(self): |
| 139 | f = tempfile.NamedTemporaryFile(mode='w+', delete=False) |
| 140 | f.close() |
| 141 | self.net.save(f.name) |
| 142 | net_file = simple_net_file(self.num_output) |
| 143 | # Test legacy constructor |
| 144 | # should print deprecation warning |
| 145 | caffe.Net(net_file, f.name, caffe.TRAIN) |
| 146 | # Test named constructor |
| 147 | net2 = caffe.Net(net_file, caffe.TRAIN, weights=f.name) |
| 148 | os.remove(net_file) |
| 149 | os.remove(f.name) |
| 150 | for name in self.net.params: |
| 151 | for i in range(len(self.net.params[name])): |
| 152 | self.assertEqual(abs(self.net.params[name][i].data |
| 153 | - net2.params[name][i].data).sum(), 0) |
| 154 | |
| 155 | def test_save_hdf5(self): |
| 156 | f = tempfile.NamedTemporaryFile(mode='w+', delete=False) |
nothing calls this directly
no test coverage detected