You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi zhenghao,
the train/eval mode of the fcos model is written in your code [path/model/fcos.py], however, in the line 26
'super().train(mode=True)', the model mode is fixed as training mode.
That leads to some unexpected inference results when doing inference by set model to eval(). So, I suggest
modify 'super().train(mode=True)' to 'super().train(mode=mode)'.
Otherwise, the dropout/bn is not locked in the inference procedure. At the same time, the dropout/bn status
can be affected by the [path/model/config.py] configurations: freeze_stage1/freeze_bn
The text was updated successfully, but these errors were encountered:
Hi zhenghao,
the train/eval mode of the fcos model is written in your code [path/model/fcos.py], however, in the line 26
'super().train(mode=True)', the model mode is fixed as training mode.
That leads to some unexpected inference results when doing inference by set model to eval(). So, I suggest
modify 'super().train(mode=True)' to 'super().train(mode=mode)'.
Otherwise, the dropout/bn is not locked in the inference procedure. At the same time, the dropout/bn status
can be affected by the [path/model/config.py] configurations: freeze_stage1/freeze_bn
The text was updated successfully, but these errors were encountered: