marcellacornia / sam

Predicting Human Eye Fixations via an LSTM-based Saliency Attentive Model. IEEE Transactions on Image Processing (2018)
https://ieeexplore.ieee.org/document/8400593
MIT License
205 stars 76 forks source link

Retraining the models #37

Open samzabihi opened 4 years ago

samzabihi commented 4 years ago

Hi, Thanks for open sourcing the code :) I successfully run your pretrained models on the Salicon dataset. Now I want to retrain these models in order to achieve the wights of pretrained models by myself. does current training setup lead to the same wights or I'm supposed to change the learning rate and optimizer?