I used adam optimizer In this model training, And keep the low learning way.

​

Firstly, I set up a data loader for loading coco data become a tensor format, that can easily process the training.

​

Secondly, put the image tensor into model and it will output a predict tensor.

Let prediction and Labels into loss function to calculate the loss and backward whole network.

​

run these in 2 epoch.

​