add bughook in train_net.py

2021-10-16 20:56:01 +08:00 · 2021-10-16 20:56:01 +08:00 · 786c442391
parent d87f4a8419
commit 786c442391
2 changed files with 1 additions and 0 deletions
--- a/fastreid/utils/bughook.py
+++ b/fastreid/utils/bughook.py
--- a/projects/FastShoe/train_net.py
+++ b/projects/FastShoe/train_net.py
@ -13,6 +13,7 @@ sys.path.append('.')
 from fastreid.config import get_cfg
 from fastreid.engine import default_argument_parser, default_setup, launch
 from fastreid.utils.checkpoint import Checkpointer, PathManager
+from fastreid.utils import bughook

 from fastshoe import PairTrainer