Unpickle GPU-trained Autoencoder on CPU-only device #1
Description
The following currently does not work:
- train an autoencoder on a gpu
- store the model in pickel format (as is done here)
- load the model on non-gpu machine
According to this issue it should be fixed in skorch. There might be an issue with the current implementation on my side that causes a weird bug.