Skip to content
This repository was archived by the owner on Jan 22, 2025. It is now read-only.
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions d2go/runner/default_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@
from d2go.trainer.fsdp import get_grad_scaler
from d2go.trainer.helper import parse_precision_from_string
from d2go.utils.flop_calculator import attach_profilers
from d2go.utils.helper import D2Trainer, TensorboardXWriter
from d2go.utils.helper import D2Trainer, TensorBoardXWriter
from d2go.utils.misc import get_tensorboard_log_dir
from d2go.utils.visualization import DataLoaderVisWrapper, VisualizationEvaluator
from detectron2.checkpoint import DetectionCheckpointer, PeriodicCheckpointer
Expand Down Expand Up @@ -72,7 +72,7 @@

@lru_cache()
def _get_tbx_writer(log_dir, window_size=20):
ret = TensorboardXWriter(log_dir, window_size=window_size)
ret = TensorBoardXWriter(log_dir, window_size=window_size)
ALL_TB_WRITERS.append(ret)
return ret

Expand Down
4 changes: 2 additions & 2 deletions d2go/utils/helper.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
PascalVOCDetectionEvaluator,
SemSegEvaluator,
)
from detectron2.utils.events import TensorboardXWriter
from detectron2.utils.events import TensorBoardXWriter
from mobile_cv.common.misc.oss_utils import fb_overwritable

T = TypeVar("T")
Expand All @@ -33,7 +33,7 @@
"run_once",
"retryable",
"get_dir_path",
"TensorboardXWriter", # TODO: move to D2Go's vis utils if needed
"TensorBoardXWriter", # TODO: move to D2Go's vis utils if needed
"D2Trainer", # TODO: move to trainer folder
]

Expand Down