search

scu-zjz / IMDLBenCo

[NeurIPS'24 Spotlight] A comprehensive benchmark & codebase for Image manipulation detection/localization.
https://scu-zjz.github.io/IMDLBenCo-doc
Creative Commons Attribution 4.0 International
77 stars 12 forks source link

modify train.py for mvss #4

Closed DSLJDI closed 5 months ago

DSLJDI commented 5 months ago

You need to modify model.eval() in the tester.py file to model.train() or not use model.eval() at all. Also, you need to set find_unused_parameters to True in the train.py file.

SunnyHaze commented 5 months ago

Solved with: https://github.com/scu-zjz/IMDLBenCo/commit/f41b22d97ab52b4b69c6e5cee250023727302d5c