Adapt losses to automatically use correct device

jku-vds-lab / paradime

A framework for parametric dimensionality reduction

MIT License

9 stars 3 forks source link

Adapt losses to automatically use correct device #23

Closed einbandi closed 2 years ago

einbandi commented 2 years ago

Device-related errors are fixed in most places, but they remain a problem with the current loss implementation. The losses should know whether the model of a ParametricDR lives on CPU or GPU and move the .sub output accordingly.

einbandi commented 2 years ago

This is mostly done. There remains a RuntimeError when a model parameter is used inside a transform, but I couldn't figure out yet if this issue is connected with the loss modifications.

einbandi commented 2 years ago

Issue is fixed, the RuntimeError is caused by an in-place operation in the StudentTTransform.