机器学习天坑总结篇(TensorFlow)

1. 在Jupiter notebook中,如果想要给自己的模型(输出tensor)调用自定义的function(例如dice_score)或者要输出在tf的可视化界面上时,有时需要将tensor转化为numpy类型,但是发现Tensor.numpy()这个函数用了会报错:AttributeError: Tensor has no attribute ' numpy'

此时可以在代码最前方加入。

import tensorflow as tf
tf.config.run_functions_eagerly(True)

如果想要输出在.format后的大括号里,大括号里的数据类型直接不填就可以了,默认会输出一串带有tensor类型变量基本信息的内容。

2. 忘了,等待更新(在作CV的时候,能用Pytorch就尽量用torch,TensorFlow真的不大办,在很多设计上都不如torch简洁而且好用,就这个numpy转类型的问题就头疼了我好些天,很多零零散散烦人想小问题用torch都解决了,现在网上很多模型的源码也是用torch写的)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

张北海。

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值