deepdrivemd.selection.latest.config

pydantic settings deepdrivemd.selection.latest.config.LatestCheckpointConfig

Config for selecting the latest model checkpoint.

Show JSON schema

{
   "title": "LatestCheckpointConfig",
   "description": "Config for selecting the latest model checkpoint.",
   "type": "object",
   "properties": {
      "experiment_directory": {
         "title": "Experiment Directory",
         "default": "set_by_deepdrivemd",
         "env_names": "{'experiment_directory'}",
         "type": "string",
         "format": "path"
      },
      "stage_idx": {
         "title": "Stage Idx",
         "default": 0,
         "env_names": "{'stage_idx'}",
         "type": "integer"
      },
      "task_idx": {
         "title": "Task Idx",
         "default": 0,
         "env_names": "{'task_idx'}",
         "type": "integer"
      },
      "output_path": {
         "title": "Output Path",
         "default": "set_by_deepdrivemd",
         "env_names": "{'output_path'}",
         "type": "string",
         "format": "path"
      },
      "node_local_path": {
         "title": "Node Local Path",
         "default": "set_by_deepdrivemd",
         "env_names": "{'node_local_path'}",
         "type": "string",
         "format": "path"
      },
      "retrain_freq": {
         "title": "Retrain Freq",
         "default": 1,
         "env_names": "{'retrain_freq'}",
         "type": "integer"
      },
      "checkpoint_dir": {
         "title": "Checkpoint Dir",
         "default": "checkpoint",
         "env_names": "{'checkpoint_dir'}",
         "type": "string"
      },
      "checkpoint_suffix": {
         "title": "Checkpoint Suffix",
         "default": ".pt",
         "env_names": "{'checkpoint_suffix'}",
         "type": "string"
      }
   }
}

Config

• extra: str = allow

Fields

• checkpoint_dir (str)

• checkpoint_suffix (str)

• retrain_freq (int)

field checkpoint_dir: str = 'checkpoint'

field checkpoint_suffix: str = '.pt'

field retrain_freq: int = 1