| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -43,6 +43,16 @@ def _Net_blob_loss_weights(self): | |||
| 43 | 43 | self._blob_loss_weights)) | |
| 44 | 44 | return self._blob_loss_weights_dict | |
| 45 | 45 | ||
| 46 | + @property | ||
| 47 | + def _Net_layer_dict(self): | ||
| 48 | + """ | ||
| 49 | + An OrderedDict (bottom to top, i.e., input to output) of network | ||
| 50 | + layers indexed by name | ||
| 51 | + """ | ||
| 52 | + if not hasattr(self, '_layer_dict'): | ||
| 53 | + self._layer_dict = OrderedDict(zip(self._layer_names, self.layers)) | ||
| 54 | + return self._layer_dict | ||
| 55 | + | ||
| 46 | 56 | ||
| 47 | 57 | @property | |
| 48 | 58 | def _Net_params(self): | |
@@ -321,6 +331,7 @@ def get_id_name(self): | |||
| 321 | 331 | # Attach methods to Net. | |
| 322 | 332 | Net.blobs = _Net_blobs | |
| 323 | 333 | Net.blob_loss_weights = _Net_blob_loss_weights | |
| 334 | + Net.layer_dict = _Net_layer_dict | ||
| 324 | 335 | Net.params = _Net_params | |
| 325 | 336 | Net.forward = _Net_forward | |
| 326 | 337 | Net.backward = _Net_backward | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -60,6 +60,13 @@ def test_memory(self): | |||
| 60 | 60 | for bl in blobs: | |
| 61 | 61 | total += bl.data.sum() + bl.diff.sum() | |
| 62 | 62 | ||
| 63 | + def test_layer_dict(self): | ||
| 64 | + layer_dict = self.net.layer_dict | ||
| 65 | + self.assertEqual(list(layer_dict.keys()), list(self.net._layer_names)) | ||
| 66 | + for i, name in enumerate(self.net._layer_names): | ||
| 67 | + self.assertEqual(layer_dict[name].type, | ||
| 68 | + self.net.layers[i].type) | ||
| 69 | + | ||
| 63 | 70 | def test_forward_backward(self): | |
| 64 | 71 | self.net.forward() | |
| 65 | 72 | self.net.backward() | |
| Back | FazBrowse Home | New Git URL |
0 commit comments