bcmi / DIRL-Inharmonious-Region-Localization

[ICME2021]The first work on Deep Inharmonious Region Localization, which can help image harmonization in an adversarial way.
24 stars 1 forks source link

Accuracy drops on different system (CPU and on GPU both) #2

Closed alishan2040 closed 3 years ago

alishan2040 commented 3 years ago

Hi,

Thanks for wonderful work and implementation. I've a question about the model behavior. I trained the model on cloud using a good GPU. The f1 score is good (0.7) which is acceptable for me. When I try to run the trained weights (encoder and decoder) on different System (CPU only and GPU only too). The performance drops significantly and the model is not giving same segmentation results as compared to the system where it was trained. Can you comment on this issue, why the performance decreasing on different system???

Thanks! Shan