top of page
Model Train
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.

bottom of page