Out of memory error

[c1arocque]

Hello

XXXXXXXX-deskcomputer:~/Projects/photo_restoration$ python run.py --input_folder /home/XXXXXX/Videos/08281965BC/Test/ --output_folder /home/XXXXXX/Videos/08281965BC/ColorCorrected/ --GPU 0 --HR

Running Stage 1: Overall restoration
Mapping: You are using the mapping model without global restoration.
Traceback (most recent call last):
File "test.py", line 102, in
model.initialize(opt)
File "/home/XXXXXX/Projects/photo_restoration/Global/models/mapping_model.py", line 147, in initialize
self.netG_A.cuda(opt.gpu_ids[0])
File "/home/XXXXXX/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 905, in cuda
return self._apply(lambda t: t.cuda(device))
File "/home/XXXXXX/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 797, in _apply
module._apply(fn)
File "/home/XXXXXX/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 797, in _apply
module._apply(fn)
File "/home/XXXXXX/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 820, in _apply
param_applied = fn(param)
File "/home/XXXXXX/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 905, in
return self._apply(lambda t: t.cuda(device))
RuntimeError: CUDA error: out of memory
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.

Traceback (most recent call last):
File "run.py", line 102, in
for x in os.listdir(stage_1_results):
FileNotFoundError: [Errno 2] No such file or directory: '/home/XXXXXX/Videos/08281965BC/ColorCorrected/stage_1_restore_output/restored_image'

System Details:

GPU NVIDIA GeForce GT 1030

NVidia driver 525.125.06

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

Description: Ubuntu 20.04.6 LTS
Release: 20.04
Codename: focal

16 Gb ram

Architecture: x86_64
CPU op-mode(s): 32-bit, 64-bit
Byte Order: Little Endian
Address sizes: 48 bits physical, 48 bits virtual
CPU(s): 8
On-line CPU(s) list: 0-7
Thread(s) per core: 2
Core(s) per socket: 4
Socket(s): 1
NUMA node(s): 1
Vendor ID: AuthenticAMD
CPU family: 21
Model: 2
Model name: AMD FX(tm)-8350 Eight-Core Processor
Stepping: 0
Frequency boost: enabled
CPU MHz: 1400.000

5.3 MiB PNG file
Input file attached.

I'm willing to test code if asked.

Thank You for this project. I'm hoping it will be work for my 8mm telecine (personal home videos).

Regards

Chris