"vscode:/vscode.git/clone" did not exist on "4ee043cd1ce5120864771bfb4817a1e5865f78c4"
mnist_eager to use GPU if not specified otherwise (#4216)
Update mnist_eager.py to use GPU if available and not specified otherwise. At the moment it uses the CPU even if a GPU is available, since "if flags_obj.no_gpu or tf.test.is_gpu_available():" in "def run_mnist_eager(flags_obj):" evaluates to True if a GPU is available.
Showing
Please register or sign in to comment