title: Federated Learning for Image Captioning
body: <p>I'm relatively new to tensorflow, I've read a book about machine learning and I've took Udacity's course. I've been tasked with recreating the Image Captioning Model found on this tutorial <a href="https://www.tensorflow.org/tutorials/text/image_captioning" rel="nofollow noreferrer">https://www.tensorflow.org/tutorials/text/image_captioning</a> using the Federated Learning library that tensorflow offers, so far everything is easy and I've managed to understand what I'm supposed to do until I reached the part of designing the model. In the federated learning tutorial (<a href="https://www.tensorflow.org/federated/tutorials/federated_learning_for_image_classification" rel="nofollow noreferrer">https://www.tensorflow.org/federated/tutorials/federated_learning_for_image_classification</a>), In designing the variables part, how am i supposed to know the shape of the weights and bias?</p>
  
  <p>Sorry if this is an obvious question but i've been struggling with shapes for quite sometime now, Thank you in advance! :)</p>
  
  <p>EDIT: Sorry, I forgot to add this part to the question: How am I supposed to know the variables of my model? I know that the weights and bias are the base of the model along with the loss and the accuracy is for me to know how good the model performs, but is there anything else I need to know?</p>