33 lines
824 B
Python
33 lines
824 B
Python
|
|
class Stereotype:
|
|
DATASET:str = "dataset"
|
|
MODULE:str = "module"
|
|
PIPELINE:str = "pipeline"
|
|
RUNNER:str = "runner"
|
|
FACTORY:str = "factory"
|
|
EVALUATION_METHOD:str = "evaluation_method"
|
|
LOSS_FUNCTION:str = "loss_function"
|
|
|
|
class Mode:
|
|
TRAIN:str = "train"
|
|
TEST:str = "test"
|
|
EVALUATION:str = "evaluation"
|
|
|
|
class Direcotry:
|
|
CHECKPOINT_DIR_NAME: str = 'checkpoints'
|
|
TENSORBOARD_DIR_NAME: str = 'tensorboard'
|
|
LOG_DIR_NAME: str = 'log'
|
|
RESULT_DIR_NAME: str = 'results'
|
|
|
|
class TensorBoard:
|
|
SCALAR: str = "scalar"
|
|
IMAGE: str = "image"
|
|
POINT: str = "point"
|
|
|
|
class LogType:
|
|
INFO:str = "info"
|
|
ERROR:str = "error"
|
|
WARNING:str = "warning"
|
|
SUCCESS:str = "success"
|
|
DEBUG:str = "debug"
|
|
TERMINATE:str = "terminate" |