This repository was archived by the owner on Jan 22, 2025. It is now read-only.
File tree 2 files changed +4
-4
lines changed
2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 39
39
from d2go .trainer .fsdp import get_grad_scaler
40
40
from d2go .trainer .helper import parse_precision_from_string
41
41
from d2go .utils .flop_calculator import attach_profilers
42
- from d2go .utils .helper import D2Trainer , TensorboardXWriter
42
+ from d2go .utils .helper import D2Trainer , TensorBoardXWriter
43
43
from d2go .utils .misc import get_tensorboard_log_dir
44
44
from d2go .utils .visualization import DataLoaderVisWrapper , VisualizationEvaluator
45
45
from detectron2 .checkpoint import DetectionCheckpointer , PeriodicCheckpointer
72
72
73
73
@lru_cache ()
74
74
def _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 )
76
76
ALL_TB_WRITERS .append (ret )
77
77
return ret
78
78
Original file line number Diff line number Diff line change 21
21
PascalVOCDetectionEvaluator ,
22
22
SemSegEvaluator ,
23
23
)
24
- from detectron2 .utils .events import TensorboardXWriter
24
+ from detectron2 .utils .events import TensorBoardXWriter
25
25
from mobile_cv .common .misc .oss_utils import fb_overwritable
26
26
27
27
T = TypeVar ("T" )
33
33
"run_once" ,
34
34
"retryable" ,
35
35
"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
37
37
"D2Trainer" , # TODO: move to trainer folder
38
38
]
39
39
You can’t perform that action at this time.
0 commit comments