Description
It seems the Scattering2D from Torch can't be serialized. I need this because I create several Scattering2D with pretty big dimensions (images with shapes up to 4k) - which is a time consuming process. So I would like the opportunity to be able to serialize the created Scattering2D object (actually the coefficients) so that I can apply to my images.
Steps to reproduce:
import torch
from kymatio.torch import Scattering2D
import pickle
J = 3
M, N = 32, 32
file = open('serializeScattering2DObject', 'wb')
pickle.dump(S, file)
Expected results:
the serialization should be done.
Actual results:
Error is thrown - "_pickle.PicklingError: Can't pickle <class 'kymatio.torch.ScatteringTorch2D'>: attribute lookup ScatteringTorch2D on kymatio.torch failed"