R session Aborted

[chri002]

I have tried to follow the basic torch guide: Guess the correlation; but whenever I get to the point of running the R training, an error ‘R session aborted’ returns.

[sebffischer]

You should provide the code you are trying to run as it is otherwise hard to help you.

[chri002]

The tutorial, and code, that I followed is "guess the correlation" from "https: //torch.mlverse.org /start /guess_the_correlation /".
The error message is created when executing the following command:
fitted <- net %>%
setup(
loss = function(y_hat, y_true) nnf_mse_loss(y_hat, y_true$unsqueeze(2)),
optimizer = optim_adam
) %>%
fit(train_dl, epochs = 10, valid_data = test_dl)

The other commands seem to work, I can run an inference, build the model, move it between RAM and VRAM. But when I run the training it seems to collapse.
I hope it can be resolved.