keras-surgeon icon indicating copy to clipboard operation
keras-surgeon copied to clipboard

Published 20 hours ago • verified publisher icon BenWhetton

Is it feasible to implement it on 3D Conv Layer

Open momo1986 opened this issue 4 years ago • 1 comments

Hello, I have trained a 3D layer for image-sequence classificiation problem, the model has over 100MB parameters. I want to do pruning, how can I apply your pruning API? If it is only suitable for 2D situation, does it mean I should rewrite APIs for 3D？ Thanks & Regards！

momo1986 avatar Sep 10 '19 02:09 momo1986

Yes it supports 3D convnet.

FBehrad avatar Jul 15 '21 14:07 FBehrad