Constants for common filenames
To retrieve the config file of a model in a folder checkpoint
, we currently use the command
model_config = os.path.join(checkpoint, "model_config.yml")
It would be good to have a constant that records what the standard name for this type of file is:
model_config = os.path.join(checkpoint, CaBRNet.MODEL_CONFIG_FILENAME)
(or similar). Also applies to "model_state.pth"
, "dataset.yml"
, "training.yml"
, "visualization.yml"
, and the "final"
and "best"
checkpoint folders.