We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 08a6581 commit 12b6875Copy full SHA for 12b6875
Keras-Fine-Tuning/keras-finetune-vgg.ipynb
@@ -34,16 +34,7 @@
34
"random.seed(seed)\n",
35
"tf.random.set_seed(seed)\n",
36
"np.random.seed(seed)\n",
37
- "os.environ['TF_DETERMINISTIC_OPS'] = '1'\n",
38
- "\n",
39
- "is_use_gpu = True\n",
40
- "if is_use_gpu:\n",
41
- " gpu_devices = tf.config.list_physical_devices('GPU')\n",
42
- " tf.config.experimental.set_visible_devices(gpu_devices[0], 'GPU')\n",
43
- " tf.config.experimental.set_memory_growth(gpu_devices[0], True)\n",
44
- " os.environ['TF_USE_CUDNN'] = '1'\n",
45
- "else:\n",
46
- " os.environ[\"CUDA_VISIBLE_DEVICES\"] =\"-1\""
+ "os.environ['TF_DETERMINISTIC_OPS'] = '1'"
47
]
48
},
49
{
0 commit comments