机器学习天坑总结篇(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写的)

猜你喜欢

转载自blog.csdn.net/m0_59056870/article/details/126293194