search

hkchengrex / XMem

[ECCV 2022] XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Model
https://hkchengrex.com/XMem/
MIT License
1.72k stars 191 forks source link

Why enable xmem.eval() during training #93

Closed jingliang95 closed 1 year ago

jingliang95 commented 1 year ago

Thanks for your work.

In your code, during the training (https://github.com/hkchengrex/XMem/blob/083698bbb4c5ac0ffe1a8923a6c313de46169983/model/trainer.py#L220), the model is also in eval mode? Why is it like this?

hkchengrex commented 1 year ago

To freeze batch norm following STM.