(self)
| 133 | os.remove(net_file) |
| 134 | |
| 135 | def test_parameter(self): |
| 136 | net_file = parameter_net_file() |
| 137 | net = caffe.Net(net_file, caffe.TRAIN) |
| 138 | # Test forward and backward |
| 139 | net.forward() |
| 140 | net.backward() |
| 141 | layer = net.layers[list(net._layer_names).index('layer')] |
| 142 | self.assertEqual(layer.blobs[0].data[0], 0) |
| 143 | self.assertEqual(layer.blobs[0].diff[0], 1) |
| 144 | layer.blobs[0].data[0] += layer.blobs[0].diff[0] |
| 145 | self.assertEqual(layer.blobs[0].data[0], 1) |
| 146 | |
| 147 | # Test saving and loading |
| 148 | h, caffemodel_file = tempfile.mkstemp() |
| 149 | net.save(caffemodel_file) |
| 150 | layer.blobs[0].data[0] = -1 |
| 151 | self.assertEqual(layer.blobs[0].data[0], -1) |
| 152 | net.copy_from(caffemodel_file) |
| 153 | self.assertEqual(layer.blobs[0].data[0], 1) |
| 154 | os.remove(caffemodel_file) |
| 155 | |
| 156 | # Test weight sharing |
| 157 | net2 = caffe.Net(net_file, caffe.TRAIN) |
| 158 | net2.share_with(net) |
| 159 | layer = net.layers[list(net2._layer_names).index('layer')] |
| 160 | self.assertEqual(layer.blobs[0].data[0], 1) |
| 161 | |
| 162 | os.remove(net_file) |
| 163 | |
| 164 | def test_phase(self): |
| 165 | net_file = phase_net_file() |
nothing calls this directly
no test coverage detected