cuda error

[joeliuz6]

/home/zliu106/.local/lib/python3.9/site-packages/pytorch_influence_functions/influence_function.py:69: UserWarning: Implicit dimension choice for log_softmax has been deprecated. Change the call to include dim=X as an argument.
y = torch.nn.functional.log_softmax(y)
Traceback (most recent call last):
File "/gpfs/fs1/home/zliu106/Documents/influence_function/influence.py", line 109, in
influences, harmful, helpful = ptif.calc_img_wise(config, model, trainloader, testloader)
File "/home/zliu106/.local/lib/python3.9/site-packages/pytorch_influence_functions/calc_influence_function.py", line 460, in calc_img_wise
influence, harmful, helpful, _ = calc_influence_single(
File "/home/zliu106/.local/lib/python3.9/site-packages/pytorch_influence_functions/calc_influence_function.py", line 315, in calc_influence_single
s_test_vec = calc_s_test_single(model, z_test, t_test, train_loader,
File "/home/zliu106/.local/lib/python3.9/site-packages/pytorch_influence_functions/calc_influence_function.py", line 93, in calc_s_test_single
s_test_vec_list.append(s_test(z_test, t_test, model, train_loader,
File "/home/zliu106/.local/lib/python3.9/site-packages/pytorch_influence_functions/influence_function.py", line 43, in s_test
x, t = x.cuda(), t.cuda()
RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.