联合学习的图像字幕

时间:2020-03-24 21:44:19

标签: tensorflow tensorflow2.0 tensorflow-federated

我对tensorflow还是比较陌生,我读了一本关于机器学习的书,并且上过Udacity的课程。我的任务是使用tensorflow提供的联合学习库重新创建本教程https://www.tensorflow.org/tutorials/text/image_captioning上的图像字幕模型,到目前为止,一切都很容易,并且我已经设法理解了我应该做的事情,直到我达到了设计模型的一部分。在联合学习教程(https://www.tensorflow.org/federated/tutorials/federated_learning_for_image_classification)中,在设计变量部分时,我应该如何知道权重和偏差的形状?

很抱歉,如果这是一个显而易见的问题,但是我已经有一段时间在形状方面苦苦挣扎了,谢谢! :)

编辑:对不起,我忘了将此部分添加到问题中:我应该如何知道模型的变量?我知道权重和偏差是模型以及损失的基础,准确性是我知道模型性能的好坏,但是我还需要知道什么吗?

1 个答案:

答案 0 :(得分:0)

模型的权重和偏差的形状由建模者/实验(您:D)选择。有时这被称为“挑选模型架构”,并且没有灵丹妙药。该问题已在How to decide neural network architecture?

中得到了广泛的回答。