Brady Williamson

icon company University of Cincinnati icon location Cincinnati, OH icon location Professor in the Department of Radiology at the University of Cincinnati

Results 5 comments of Brady Williamson

Model Trains on CPU but not on GPU - TensorDot/MatMul error

Hi, Just wanted to add some details to this to see if I can get a fix. I think through a few more hours of trial and error and searching...

Model Trains on CPU but not on GPU - TensorDot/MatMul error

Thanks for the response @tilakrayal and for the second @markemus. I confirmed this issue by training the network with a batch size of 1 with no issues, as the matrix...

Model Trains on CPU but not on GPU - TensorDot/MatMul error

@tilakrayal Thanks for the link. I am not sure what to try per se. I read through and it looks like this change was committed to the master branch of...

Model Trains on CPU but not on GPU - TensorDot/MatMul error

@tilakrayal I apologize for the delay. I updated TF and am still getting the same error. My current version of TF is 2.11.0-dev20220928

Model Trains on CPU but not on GPU - TensorDot/MatMul error

@arvindrajan92 That fixed it!! Thanks very much for your suggestion.