Can I use the CPU for training

jpWang / LiLT

Official PyTorch implementation of LiLT: A Simple yet Effective Language-Independent Layout Transformer for Structured Document Understanding (ACL 2022)

MIT License

335 stars 40 forks source link

Can I use the CPU for training #19

Closed T00ler closed 1 year ago

T00ler commented 1 year ago

Can I use the CPU for training, as my current graphics card does not have enough video memory (Error Alert: [RuntimeError: CUDA out of memory]), and I just want to make sure that it will run properly on my computer. :(

jpWang commented 1 year ago

Hi, I haven't tried to train this repo with CPU, but I think it will be very, very slow...