基于tensorflow2实现卷积神经网络

本文介绍了如何使用TensorFlow2的API构建一个卷积神经网络,包括数据集的分布、训练过程中的问题及解决办法。在训练中遇到梯度消失问题,可能的原因是数据未归一化或者卷积层缺少激活函数。通过调整这两个方面,可以有效解决此问题。
摘要由CSDN通过智能技术生成

利用tensorflow2中的api实现一个简单的卷积神经网络,完成梯度下降的操作并绘制训练集和测试集准确率曲线。
数据集在这里
数据分布:训练集数量为209,测试集数量为50

import numpy as np
import matplotlib.pyplot as plt
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers

在之前的编程过程中,始终会发生调用不了This is probably because cuDNN failed to initializeThis is probably because GEMM failed to initialize等错误,通过查询相关问题,最有可能和tensorflow总是自动申请全部显存有关,所以有相关问题的同学们可以在程序开头增加下面的代码,通过设定Tensorflow申请显存的占比来解决这个问题(因为网上能找到的都是tensorflow1版本中的代码,所以还需要用到config=tf.compat.v1.ConfigProto()让代码能成功运行)。

config=tf.compat.v1.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5
sess =tf.compat.v1.Session(config=config)
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值