随笔分类 -  tensorflow2

摘要:TFRecord文件训练速度调优 问题描述: 背景: tfrecord文件由spark生成: res_df.write.format("tfrecords").option("recordType","Example")mode("overwrite").save("path/") 大约有90个特征 阅读全文
posted @ 2023-09-12 15:10 风和雨滴 阅读(205) 评论(0) 推荐(0)
摘要:Blas xGEMV launch failed : a.shape=[1,2048000,50], b.shape=[1,50,1], m=2048000, n=1, k=50 问题描述:使用tensorflow2运行模型时报错: 2 root error(s) found. (0) Intern 阅读全文
posted @ 2023-09-12 14:47 风和雨滴 阅读(155) 评论(0) 推荐(0)
摘要:一般情况下我们使用一个batch的数据计算出一个平均的loss,再使用这个loss反向传播求得模型参数的梯度并更新,例如: loss = tf.reduce_mean(y - y_pred) # 文中y和y_pred的shape均为:[b, 1],其中b是batch_size. 这种情况比较好理解, 阅读全文
posted @ 2021-11-07 13:51 风和雨滴 阅读(1869) 评论(0) 推荐(1)
摘要:踩坑事项: 一定要将for循环的次数在构建层时就传入__init__(),而不是和inputs一起使用call传入!特别是每轮循环都需要tf.concat()新张量的时候。否则tf构建动态图无法计算循环后数据的维度报错! 阅读全文
posted @ 2021-11-01 17:21 风和雨滴 阅读(269) 评论(0) 推荐(0)