The number of channels of the input layer is currently effectively hardcoded to be 16. This makes it a template parameter of Wavenet_Model to allow other architectures.
It may be possible to extract this from the first layer of the network, but if so it is beyond my C++ template skills...
The number of channels of the input layer is currently effectively hardcoded to be 16. This makes it a template parameter of Wavenet_Model to allow other architectures.
It may be possible to extract this from the first layer of the network, but if so it is beyond my C++ template skills...