search

cvxgrp / pymde

Minimum-distortion embedding with PyTorch
https://pymde.org
Apache License 2.0
536 stars 27 forks source link

Fix anchored preserve_neighbors on cuda #46

Closed frederikschubert closed 2 years ago

frederikschubert commented 2 years ago

Currently, preserve_neighbors might fail with an Anchored constraint on cuda devices. This PR fixes the tests to showcase this failure and prevents it.