-
- Downloads
A temporary fix for issue #36 (#48)
* A temporary fix for issue 36 The thinning code assumes that the sgraph it is using is not data-parallel, because it (currently) accesses the layer-name keys using a "normalized" name ("module." is removed). The bug is that in thinning.py#L73 we create a data_parallel=True model; and then give it to sgraph. But in other places thinning code uses "normalized" keys. For example in thinning.py#L264. The temporary fix configures data_parallel=False in thinning.py#L73. A long term solution should have SummaryGraph know how to handle both parallel and not-parallel models. This can be done by having SummaryGraph convert layer-names it receives in the API to data_parallel=False using normalize_layer_name. When returning results, use the de-normalized format. * Fix the documentation error from issue 36 * Move some logs to debug and show in logging.conf how to enable DEBUG logs.
Showing
- apputils/model_summaries.py 1 addition, 0 deletionsapputils/model_summaries.py
- distiller/thinning.py 3 additions, 1 deletiondistiller/thinning.py
- docs-src/docs/model_zoo.md 4 additions, 3 deletionsdocs-src/docs/model_zoo.md
- docs/index.html 1 addition, 1 deletiondocs/index.html
- docs/model_zoo/index.html 4 additions, 3 deletionsdocs/model_zoo/index.html
- docs/search/search_index.json 359 additions, 359 deletionsdocs/search/search_index.json
- docs/sitemap.xml 13 additions, 13 deletionsdocs/sitemap.xml
- examples/classifier_compression/logging.conf 8 additions, 0 deletionsexamples/classifier_compression/logging.conf
Loading
Please register or sign in to comment