Open YangYangGirl opened 5 years ago
I wrote the poink-linking-network based on your code,but the loss looked strange which leaped after every epoch.So I had took a close look out at your loss,I found that there also was a relatively weak leaping after every epoch.I felt very strange about it. https://github.com/YangYangGirl/point-linking-network
The leap is normal because the current repository only supports batch_size = 1 training, it means it randomly select out one image instead of a mini batch that includes multiple images, this randomness causes the leap after every epoch, if you change the batch_size to 8 our 16, the leap will disappear.