Closed luigibonati closed 1 year ago
List of changes:
Both the mlcolvar/tests/test_cvs_multitask_multitask.py and DeepTDA notebook fail for tensors on different devices (in local, by default GPU is used if available)
@EnricoTrizio I think the tests are failing on GPU when TDA loss is used:
# compute loss function contributes for class i
> loss_centers[i] = alpha*(mu - target_centers[i]).pow(2)
E RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
@andrrizzi @EnricoTrizio I am merging this. Note that you might have to reinstall the library The last thing which needs to be changed is the repo name (and then also read the docs)
Description
renaming library, lightning and other stuff