MhLiao / MaskTextSpotter

A PyTorch implementation of Mask TextSpotter
https://github.com/MhLiao/MaskTextSpotter
412 stars 94 forks source link

Whether a single GPU can run #16

Closed machine52vision closed 4 years ago

machine52vision commented 4 years ago

Whether a single GPU can run

MhLiao commented 4 years ago

@machine52vision In the inference period, it can run with a single GPU. You can train with a single GPU with smaller batch size but I have not tried that. The performance may relate to the training batch size.

phucnsp commented 4 years ago

Hello @MhLiao, Can I inference your model with CPU?