CUDA error:out of memery and other errors

[ALISA199812]

when I run the code -- python test_nas_video_process.py --quality [quality level] --data_name [dataset name] --use_cuda --load_on_memory
The error is the fig.1,and I checked the Display Card that is out of merrory.So I made the following adjustments to the code files：
1.option.py
parser.add_argument("--num_batch", type=int, default=64) to num_batch=1
2.process
MAX_FPS = 30 to MAX_FPS = 10
MAX_SEGMENT_LENGTH = 4 to MAX_SEGMENT_LENGTH = 1
3.test_nas_video_process.py
MAX_FPS = 30 to MAX_FPS = 10
MAX_SEGMENT_LENGTH = 4 to MAX_SEGMENT_LENGTH = 1

But an new error is still reported in the fig.2

It seems that the process is too much so that make it error.
Could you give me some advice？thanks

my computer is :
CUDA11.1
torch 1.9.0(pip installed)
RTX2060 6G

Looking forward to your reply!