Equalization
channel dimension is not calculated during trace time
#431
Labels
Equalization
channel dimension is not calculated during trace time
#431
There are a few issues I faced while trying to use the
Equalization
layer with AugMix in #407The channel dimension is not calculated during trace time (I believe due to the
vectorized_map
from #201). This makes it incompatible with other layers in a chain. Moreover, this is the only layer that gets traced every time it's called. It's not ideal and repeated occurrences of this layer are very inefficient.The text was updated successfully, but these errors were encountered: