This repository was archived by the owner on Jan 22, 2025. It is now read-only.
File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 3939from d2go .trainer .fsdp import get_grad_scaler
4040from d2go .trainer .helper import parse_precision_from_string
4141from d2go .utils .flop_calculator import attach_profilers
42- from d2go .utils .helper import D2Trainer , TensorboardXWriter
42+ from d2go .utils .helper import D2Trainer , TensorBoardXWriter
4343from d2go .utils .misc import get_tensorboard_log_dir
4444from d2go .utils .visualization import DataLoaderVisWrapper , VisualizationEvaluator
4545from detectron2 .checkpoint import DetectionCheckpointer , PeriodicCheckpointer
7272
7373@lru_cache ()
7474def _get_tbx_writer (log_dir , window_size = 20 ):
75- ret = TensorboardXWriter (log_dir , window_size = window_size )
75+ ret = TensorBoardXWriter (log_dir , window_size = window_size )
7676 ALL_TB_WRITERS .append (ret )
7777 return ret
7878
Original file line number Diff line number Diff line change 2121 PascalVOCDetectionEvaluator ,
2222 SemSegEvaluator ,
2323)
24- from detectron2 .utils .events import TensorboardXWriter
24+ from detectron2 .utils .events import TensorBoardXWriter
2525from mobile_cv .common .misc .oss_utils import fb_overwritable
2626
2727T = TypeVar ("T" )
3333 "run_once" ,
3434 "retryable" ,
3535 "get_dir_path" ,
36- "TensorboardXWriter " , # TODO: move to D2Go's vis utils if needed
36+ "TensorBoardXWriter " , # TODO: move to D2Go's vis utils if needed
3737 "D2Trainer" , # TODO: move to trainer folder
3838]
3939
You can’t perform that action at this time.
0 commit comments