hola5156
commented
3 weeks ago The error above occurred while processing 2160p video, while the error below occurred while processing 1080p video.
root@gpu-1075d5aec6736b6422097-1-5524:~/PerVFI# python infer_video.py -m gmflow+pervfi -data input -fps 60
Testing on Dataset: input
Running VFI method : gmflow+pervfi
TMP (temporary) Dir: /tmp/tmp5e4xc0ik
VIS (visualize) Dir: output
Building VFI model...
Done
1
0%| 0%| | 0/1 [00:02<?, ?it/s]
Traceback (most recent call last):
File "/root/PerVFI/infer_video.py", line 110, in
File "<frozen os>", line 684, in __setitem__
File "<frozen os>", line 758, in encode
TypeError: str expected, not NoneType
python infer_video.py -m gmflow+pervfi-vb -data input -fps 11 Testing on Dataset: input Running VFI method : gmflow+pervfi-vb TMP (temporary) Dir: /tmp/tmp62froo40 VIS (visualize) Dir: output Building VFI model... 2024-11-03 07:10:30.219 | INFO | models.generators.PFlowVFI_Vb:init:262 - Parameter of decoder: 751875 Done 1 0%| | 0/1 [00:01<?, ?it/s] Traceback (most recent call last): File "/output/PerVFI-main/infer_video.py", line 110, in
outs = inferRGB(*inps) # e.g., [I2]
^^^^^^^^^^^^^^^
File "/output/PerVFI-main/infer_video.py", line 84, in inferRGB tenOut = infer(*inputs, time=t) ^^^^^^^^^^^^^^^^^^^^^^
File "/output/PerVFI-main/build_models.py", line 39, in infer pred = model.inference_rand_noise(I1, I2, heat=0.3, time=time) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/envs/pervfi/lib/python3.11/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context return func(*args, kwargs) ^^^^^^^^^^^^^^^^^^^^^ File "/output/PerVFI-main/models/pipeline.py", line 61, in inference_rand_noise fflow, bflow = flows if flows is not None else self.compute_flow(img0, img1) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/local/envs/pervfi/lib/python3.11/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context return func(*args, *kwargs) ^^^^^^^^^^^^^^^^^^^^^ File "/output/PerVFI-main/models/flow_estimators/init.py", line 139, in infer results_dict = model( ^^^^^^ File "/usr/local/envs/pervfi/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(args, kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/local/envs/pervfi/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(*args, kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/output/PerVFI-main/models/flow_estimators/gmflow/gmflow.py", line 136, in forward feature0, feature1 = self.transformer(feature0, feature1, attn_num_splits=attn_splits) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/local/envs/pervfi/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, *kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/local/envs/pervfi/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(args, kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/output/PerVFI-main/models/flow_estimators/gmflow/transformer.py", line 290, in forward shifted_window_attn_mask = generate_shift_window_attn_mask( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/output/PerVFI-main/models/flow_estimators/gmflow/transformer.py", line 41, in generate_shift_window_attn_mask attn_mask = attn_mask.masked_fill(attn_mask != 0, float(-100.0)).masked_fill(attn_mask == 0, float(0.0)) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 13.84 GiB. GPU 0 has a total capacty of 23.65 GiB of which 4.46 GiB is free. Process 301132 has 19.18 GiB memory in use. Of the allocated memory 18.39 GiB is allocated by PyTorch, and 344.78 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF