search

SunnyHaze / IML-ViT

Official repository of paper “IML-ViT: Benchmarking Image manipulation localization by Vision Transformer”
MIT License
184 stars 23 forks source link

自动半精度从哪里修改 #27

Closed changchaoqun closed 1 month ago

changchaoqun commented 1 month ago

作者你好，请问amp在哪个脚本里修改，具体位置是哪

SunnyHaze commented 1 month ago

你好，感谢你对我们工作的关注。

自动半精度来自这里：

https://github.com/SunnyHaze/IML-ViT/blob/ae270c7f48f86ba06b308c8151d870e0a5a9b05c/engine_train.py#L47

changchaoqun commented 1 month ago

谢谢~，已完成修改