mapdet3d/engine/flag.py

"""Engine Flags."""

from absl import flags

from .parser import DEFINE_config_file

_CONFIG = DEFINE_config_file("config", method_name="get_config")
_GPUS = flags.DEFINE_integer("gpus", default=0, help="Number of GPUs per node")
_NODES = flags.DEFINE_integer("nodes", default=1, help="Number of nodes")
_WANDB = flags.DEFINE_bool(
    "wandb", default=False, help="If set, use Weights & Biases for logging."
)
_CKPT = flags.DEFINE_string("ckpt", default=None, help="Checkpoint path")
_RESUME = flags.DEFINE_bool("resume", default=False, help="Resume training")
_SHOW_CONFIG = flags.DEFINE_bool(
    "print-config", default=False, help="If set, prints the configuration."
)
_VIS = flags.DEFINE_bool(
    "vis",
    default=False,
    help="If set, running visualization using visualizer callback.",
)


__all__ = [
    "_CONFIG",
    "_GPUS",
    "_NODES",
    "_CKPT",
    "_RESUME",
    "_SHOW_CONFIG",
    "_WANDB",
    "_VIS",
]