modelforge/potential/utils.py:378: UserWarning: To copy construct from a tensor, it is recommended to use sourceTensor.clone().detach() or sourceTensor.clone().detach().requires_grad_(True), rather than torch.tensor(sourceTensor).
widths = torch.tensor(
Using the training script a UserWanring is shown:
This should be resolved.