Reply to InvalidArgumentError: Cannot assign a device for operation agent/VerifyFinite/CheckNumerics Hi @Derek C, @Frameroks Engineer I have run threw a similar issue, have you found a way to fix it, running on gpu? Thanks Machine Learning & AI General Jul ’22