search

scu-zjz / IMDLBenCo

A comprehensive benchmark & codebase for Image manipulation detection/localization.
https://scu-zjz.github.io/IMDLBenCo-doc
Creative Commons Attribution 4.0 International
22 stars 3 forks source link

modify train.py for mvss #4

Closed DSLJDI closed 1 month ago

DSLJDI commented 1 month 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 1 month ago

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