SMPLify-X on HICO bad images $ ./bad_image_fit4.sh

[2440:2429 0:2108] 10:16:17 Wed Jan 13 [mona@goku:pts/0 +1] ~/research/code/smplify-x
$ ./bad_image_fit4.sh
Processing: ../../data/smplify-x/BAD_DATA_FOLDER4/images/HICO_test2015_00000471.jpg
Found Trained Model: ../../data/smplify-x/vposer_v1_0/snapshots/TR00_E096.pt
/home/mona/research/code/smplify-x/smplifyx/optimizers/lbfgs_ls.py:238: UserWarning: This overload of add_ is deprecated:
	add_(Number alpha, Tensor other)
Consider using one of the following signatures instead:
	add_(Tensor other, *, Number alpha) (Triggered internally at  /pytorch/torch/csrc/utils/python_arg_parser.cpp:882.)
  p.data.add_(step_size, update[offset:offset + numel].view_as(p.data))
Camera initialization done after 1.1937
Camera initialization final loss 395.8522
Stage 000 done after 2.0787 seconds
Stage 001 done after 0.9554 seconds
Stage 002 done after 2.9617 seconds
Stage:  60%|██████████████████████████████████████████████▊                               | 3/5 [00:06<00:04,  2.03s/it]
Orientation:   0%|                                                                                | 0/2 [00:06<?, ?it/s]
Traceback (most recent call last):
  File "smplifyx/main.py", line 272, in <module>
    main(**args)
  File "smplifyx/main.py", line 245, in main
    fit_single_frame(img, keypoints[[person_id]],
  File "/home/mona/research/code/smplify-x/smplifyx/fit_single_frame.py", line 439, in fit_single_frame
    final_loss_val = monitor.run_fitting(
  File "/home/mona/research/code/smplify-x/smplifyx/fitting.py", line 175, in run_fitting
    loss = optimizer.step(closure)
  File "/home/mona/research/code/smplify-x/smplifyx/optimizers/lbfgs_ls.py", line 393, in step
    loss, flat_grad, t, ls_func_evals = _strong_Wolfe(obj_func, x_init, t, d,
  File "/home/mona/research/code/smplify-x/smplifyx/optimizers/lbfgs_ls.py", line 46, in _strong_Wolfe
    f_new, g_new = obj_func(x, t, d)
  File "/home/mona/research/code/smplify-x/smplifyx/optimizers/lbfgs_ls.py", line 392, in obj_func
    return self._directional_evaluate(closure, x, t, d)
  File "/home/mona/research/code/smplify-x/smplifyx/optimizers/lbfgs_ls.py", line 251, in _directional_evaluate
    loss = float(closure())
  File "/home/mona/research/code/smplify-x/smplifyx/fitting.py", line 246, in fitting_func
    total_loss = loss(body_model_output, camera=camera,
  File "/home/mona/venv/smplifyx/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/home/mona/research/code/smplify-x/smplifyx/fitting.py", line 434, in forward
    collision_idxs = self.search_tree(triangles)
  File "/home/mona/venv/smplifyx/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/home/mona/venv/smplifyx/lib/python3.8/site-packages/mesh_intersection/bvh_search_tree.py", line 56, in forward
    return BVHFunction.apply(triangles)
  File "/home/mona/venv/smplifyx/lib/python3.8/site-packages/torch/autograd/grad_mode.py", line 26, in decorate_context
    return func(*args, **kwargs)
  File "/home/mona/venv/smplifyx/lib/python3.8/site-packages/mesh_intersection/bvh_search_tree.py", line 38, in forward
    outputs = bvh_cuda.forward(triangles,
MemoryError: std::bad_alloc: cudaErrorMemoryAllocation: out of memory
20711/31772MB(smplifyx)