In that case you could slice the sequential model and create a new one containing only all submodules to index 32 or use a forward_hook.
1 Like
In that case you could slice the sequential model and create a new one containing only all submodules to index 32 or use a forward_hook.