how to bash train.sh

[reallllyang]

(nerfrpn) edith@edith-Legion-R9000P-ARX8:~/文档/NeRF_RPN-main/nerf_rpn$ bash train.sh

• set -e
• DATA_ROOT=/home/edith/文档/nerf_rpn_dataset/front3d_rpn_data
• python3 -u run_rpn.py --mode train --dataset_name front3d --resolution 160 --backbone_type swin_s --features_path /home/edith/文档/nerf_rpn_dataset/front3d_rpn_data/features --boxes_path /home/edith/文档/nerf_rpn_dataset/front3d_rpn_data/obb --dataset_split /home/edith/文档/nerf_rpn_dataset/front3d_rpn_data/3dfront_split.npz --save_path ./results/front3d_anchor_swins --num_epochs 200 --lr 3e-4 --weight_decay 1e-3 --log_interval 10 --eval_interval 10 --rpn_nms_thresh 0.3 --log_to_file --normalize_density --rotated_bbox --batch_size 1 --gpus 0
  [2024-02-27 21:05:04,882 INFO] Constructing model...
  [2024-02-27 21:05:06,002 INFO] Loading dataset split from /home/edith/文档/nerf_rpn_dataset/front3d_rpn_data/3dfront_split.npz.
  100%|██████████████████████████████████████| 122/122 [00:00<00:00, 27667.21it/s]
  100%|████████████████████████████████████████| 20/20 [00:00<00:00, 27077.50it/s]
  [2024-02-27 21:05:06,010 INFO] Loaded 122 training scenes, 20 validation scenes
  Traceback (most recent call last):
  File "/home/edith/文档/NeRF_RPN-main/nerf_rpn/run_rpn.py", line 695, in
  main()
  File "/home/edith/文档/NeRF_RPN-main/nerf_rpn/run_rpn.py", line 682, in main
  trainer.train_loop()
  File "/home/edith/文档/NeRF_RPN-main/nerf_rpn/run_rpn.py", line 358, in train_loop
  self.train_epoch(epoch)
  File "/home/edith/文档/NeRF_RPN-main/nerf_rpn/run_rpn.py", line 384, in train_epoch
  proposals, losses, scores = self.model(rgbsigma, boxes)
  File "/home/edith/Anaconda3/envs/nerfrpn/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
  return forward_call(*input, **kwargs)
  File "/home/edith/文档/NeRF_RPN-main/nerf_rpn/model/nerf_rpn.py", line 211, in forward
  features = list(self.backbone(mesh_tensors))
  File "/home/edith/Anaconda3/envs/nerfrpn/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
  return forward_call(*input, **kwargs)
  File "/home/edith/文档/NeRF_RPN-main/nerf_rpn/model/feature_extractor.py", line 785, in forward
  x = self.stagesi
  File "/home/edith/Anaconda3/envs/nerfrpn/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
  return forward_call(*input, **kwargs)
  File "/home/edith/Anaconda3/envs/nerfrpn/lib/python3.9/site-packages/torch/nn/modules/container.py", line 139, in forward
  input = module(input)
  File "/home/edith/Anaconda3/envs/nerfrpn/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
  return forward_call(*input, **kwargs)
  File "/home/edith/文档/NeRF_RPN-main/nerf_rpn/model/feature_extractor.py", line 644, in forward
  x = x + self.stochastic_depth(self.attn(self.norm1(x)))
  File "/home/edith/Anaconda3/envs/nerfrpn/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
  return forward_call(*input, **kwargs)
  File "/home/edith/文档/NeRF_RPN-main/nerf_rpn/model/feature_extractor.py", line 578, in forward
  return shifted_window_attention(
  File "/home/edith/文档/NeRF_RPN-main/nerf_rpn/model/feature_extractor.py", line 446, in shifted_window_attention
  qkv = F.linear(x, qkv_weight, qkv_bias)

RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)