I built a custom filter in pytorch as shown below.
When using only one cpu or gpu, the code works fine, but when using dataparallel and model dataparallel, the code does not run.
Is there any way to solve the dataparallel problem in custom filter?