分类：TensorFlow2

2022
03-17

浅谈Tensorflow2对GPU内存的分配策略

目录一、问题源起二、开发环境三、Tensorflow针对GPU内存的分配策略四、问题分析验证五、GPU分配策略分析六、扩展一、问题源起从以下的异常堆栈可以看到是BLAS程序集初始化失败，可以看到是执行MatMul的时候发生的异常，基本可以断定可能数据集太大导致memory不够用了。2021-08-1016:38:04.917501:Etensorflow/stream_executor/cuda/cuda_blas.cc:226]failedtocreatecublashandle:CUBLAS_STATUS_NOT_INITIALIZED2021-08-1... 继续阅读 >

2021
11-17

tensorflow2 自定义损失函数使用的隐藏坑

Keras的核心原则是逐步揭示复杂性，可以在保持相应的高级便利性的同时，对操作细节进行更多控制。当我们要自定义fit中的训练算法时，可以重写模型中的train_step方法，然后调用fit来训练模型。这里以tensorflow2官网中的例子来说明：importnumpyasnpimporttensorflowastffromtensorflowimportkerasx=np.random.random((1000,32))y=np.random.random((1000,1))classCustomModel(keras.Model):tf.random.set_see... 继续阅读 >

2021
08-23

教你使用TensorFlow2识别验证码

验证码是根据随机字符生成一幅图片，然后在图片中加入干扰象素，用户必须手动填入，防止有人利用机器人自动批量注册、灌水、发垃圾广告等等。数据集来源：https://www.kaggle.com/fournierp/captcha-version-2-images图片是5个字母的单词，可以包含数字。这些图像应用了噪声（模糊和一条线）。它们是200x50PNG。我们的任务是尝试制作光学字符识别算法的模型。在数据集中存在的验证码png图片，对应的标签就是图片的名字。import... 继续阅读 >

2021
02-21

TensorFlow2.0使用keras训练模型的实现

1.一般的模型构造、训练、测试流程#模型构造inputs=keras.Input(shape=(784,),name='mnist_input')h1=layers.Dense(64,activation='relu')(inputs)h1=layers.Dense(64,activation='relu')(h1)outputs=layers.Dense(10,activation='softmax')(h1)model=keras.Model(inputs,outputs)#keras.utils.plot_model(model,'net001.png',show_shapes=True)model.compile(optimizer=keras.optimizers.RMSprop(),loss=k... 继续阅读 >