Skip to content

Commit 12b6875

Browse files
author
Anastasia
committed
Minor notebook corrections
1 parent 08a6581 commit 12b6875

File tree

1 file changed

+1
-10
lines changed

1 file changed

+1
-10
lines changed

Keras-Fine-Tuning/keras-finetune-vgg.ipynb

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -34,16 +34,7 @@
3434
"random.seed(seed)\n",
3535
"tf.random.set_seed(seed)\n",
3636
"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\""
37+
"os.environ['TF_DETERMINISTIC_OPS'] = '1'"
4738
]
4839
},
4940
{

0 commit comments

Comments
 (0)