search

ohhhyeahhh / SiamGAT

Code for the paper "Graph Attention Tracking". (CVPR2021)
https://openaccess.thecvf.com/content/CVPR2021/papers/Guo_Graph_Attention_Tracking_CVPR_2021_paper.pdf
139 stars 19 forks source link

Training problem #21

Open HasilPark opened 2 years ago

HasilPark commented 2 years ago

I trained by setting the epoch parameter to 20 using got-10k dataset. However, only epochs 1 to 19 are saved. What's the problem?

Capricorn231 commented 2 years ago

When the total training data can be divisible by batch_size, the model of the 20th epoch cannot be saved. This is a small problem in our code, which will be optimized later. You can save the model of the 20th epoch by adjusting the batch_size.