对模型进行剪枝时出现错误:
Please initialize `Prune` with a supported layer. Layers should either be a `PrunableLayer` instance, or should be supported by the PruneRegistry. You passed:
方法1
1:找到prune_registry.py文件,如果是anacoda3,文件路径可能在:
1)anaconda3/envs/tensorflow2_latest_p37/lib/python3.7/site_packages/tensorflow_model_optimization/python/core/sparsity/keras/prune_registry.py
2)anaconda3\Lib\site-packages\tensorflow_model_optimization\python\core\sparsity\keras\prune_registry.py
2:打开上述文件,不建议用txt文件打开;开头添加 代码:
from tensorflow.python.keras.engine.functional import Functional
同时,在**_LAYERS_WEIGHTS_MAP **中添加:
Functional: [],<