Inference on CPU - Githubissues

hmorimitsu / ptlflow

PyTorch Lightning Optical Flow models, scripts, and pretrained weights.

Apache License 2.0

250 stars 33 forks source link

Closed GotG closed 2 years ago

GotG commented 2 years ago

Hello, how could we run inference on cpu for these optical flow models? Thanks!

hmorimitsu commented 2 years ago

Hi, they should be able to run on CPU or GPU without any changes. Are you seeing some error when running on CPU?

GotG commented 2 years ago

Hello, it was some local setup problem. It is working. Thanks!