Add `device` flag to script to run model on CPU or GPU - Githubissues

iree-org / iree-torch

Torch Frontend for IREE

Apache License 2.0

25 stars 11 forks source link

Add `device` flag to script to run model on CPU or GPU #18

Closed ramiro050 closed 2 years ago