RuntimeError

[zuiwomeirenxi]

Traceback (most recent call last):
File "C:\video_anomaly_diffusion-main\video_anomaly_diffusion-main\train_ano.py", line 371, in
main()
File "C:\video_anomaly_diffusion-main\video_anomaly_diffusion-main\train_ano.py", line 367, in main
main_objective(config, args)
File "C:\video_anomaly_diffusion-main\video_anomaly_diffusion-main\train_ano.py", line 282, in main_objective
evaluate()
File "D:\anaconda\envs\diffuser\lib\site-packages\torch\utils_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "D:\anaconda\envs\diffuser\lib\contextlib.py", line 79, in inner
return func(*args, **kwds)
File "C:\video_anomaly_diffusion-main\video_anomaly_diffusion-main\train_ano.py", line 261, in evaluate
preds_auc = auroc(gen_preds, labels,task="binary")
File "D:\anaconda\envs\diffuser\lib\site-packages\torchmetrics\functional\classification\auroc.py", line 470, in auroc
return binary_auroc(preds, target, max_fpr, thresholds, ignore_index, validate_args)
File "D:\anaconda\envs\diffuser\lib\site-packages\torchmetrics\functional\classification\auroc.py", line 174, in binary_auroc
_binary_precision_recall_curve_tensor_validation(preds, target, ignore_index)
File "D:\anaconda\envs\diffuser\lib\site-packages\torchmetrics\functional\classification\precision_recall_curve.py", line 135, in _binary_precision_recall_curve_tensor_validation
_check_same_shape(preds, target)
File "D:\anaconda\envs\diffuser\lib\site-packages\torchmetrics\utilities\checks.py", line 42, in _check_same_shape
raise RuntimeError(
RuntimeError: Predictions and targets are expected to have the same shape, but got torch.Size([138244, 512]) and torch.Size([138244]).