Why does main() train the same model twice?

NoelShin / selfmask

[CVPRW'22] Unsupervised Salient Object Detection With Spectral Cluster Voting

https://www.robots.ox.ac.uk/~vgg/research/selfmask/

MIT License

59 stars 7 forks source link

Why does main() train the same model twice? #12

Open LLL-YUE opened 1 month ago

LLL-YUE commented 1 month ago

NoelShin commented 1 month ago

Hello.

"model.train()" is not the call for training the model. It's to get some functions in the model ready for training (but one of "model.train()" is not necessary there). The actual training takes place at https://github.com/NoelShin/selfmask/blob/master/main.py#L103