《TensorFlow深度学习》学习笔记--10.卷积神经网络--1.LeNet-5实战
1. LeNet-5 实战
1. 导入相关的包
import tensorflow as tf
from tensorflow.keras import Sequential, layers, losses, optimizers
设置GPU(如果使用GPU,则需添加下面代码,如果不使用,则无需添加)
devices = tf.config.experimental.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(
原创
2021-01-20 15:42:13 ·
860 阅读 ·
0 评论