deepdrivemd.models.keras_cvae_stream.config
- pydantic settings deepdrivemd.models.keras_cvae_stream.config.KerasCVAEModelConfig
Show JSON schema
{ "title": "KerasCVAEModelConfig", "description": "Base class for specific model configs to inherit.", "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" }, "model_tag": { "title": "Model Tag", "default": "set_by_deepdrivemd", "env_names": "{'model_tag'}", "type": "string" }, "init_weights_path": { "title": "Init Weights Path", "env_names": "{'init_weights_path'}", "type": "string", "format": "path" }, "final_shape": { "title": "Final Shape", "default": [ 28, 28, 1 ], "env_names": "{'final_shape'}", "type": "array", "items": { "type": "integer" } }, "epochs": { "title": "Epochs", "default": 10, "env_names": "{'epochs'}", "type": "integer" }, "batch_size": { "title": "Batch Size", "default": 32, "env_names": "{'batch_size'}", "type": "integer" }, "split_pct": { "title": "Split Pct", "default": 0.8, "env_names": "{'split_pct'}", "type": "number" }, "shuffle": { "title": "Shuffle", "default": true, "env_names": "{'shuffle'}", "type": "boolean" }, "latent_dim": { "title": "Latent Dim", "default": 10, "env_names": "{'latent_dim'}", "type": "integer" }, "conv_layers": { "title": "Conv Layers", "default": 4, "env_names": "{'conv_layers'}", "type": "integer" }, "conv_filters": { "title": "Conv Filters", "default": [ 64, 64, 64, 64 ], "env_names": "{'conv_filters'}", "type": "array", "items": { "type": "integer" } }, "conv_filter_shapes": { "title": "Conv Filter Shapes", "default": [ [ 3, 3 ], [ 3, 3 ], [ 3, 3 ], [ 3, 3 ] ], "env_names": "{'conv_filter_shapes'}", "type": "array", "items": { "type": "array", "minItems": 2, "maxItems": 2, "items": [ { "type": "integer" }, { "type": "integer" } ] } }, "conv_strides": { "title": "Conv Strides", "default": [ [ 1, 1 ], [ 2, 2 ], [ 1, 1 ], [ 1, 1 ] ], "env_names": "{'conv_strides'}", "type": "array", "items": { "type": "array", "minItems": 2, "maxItems": 2, "items": [ { "type": "integer" }, { "type": "integer" } ] } }, "dense_layers": { "title": "Dense Layers", "default": 1, "env_names": "{'dense_layers'}", "type": "integer" }, "dense_neurons": { "title": "Dense Neurons", "default": [ 128 ], "env_names": "{'dense_neurons'}", "type": "array", "items": { "type": "integer" } }, "dense_dropouts": { "title": "Dense Dropouts", "default": [ 0.25 ], "env_names": "{'dense_dropouts'}", "type": "array", "items": { "type": "number" } }, "min_step_increment": { "title": "Min Step Increment", "default": 5000, "env_names": "{'min_step_increment'}", "type": "integer" }, "max_steps": { "title": "Max Steps", "default": 8000, "env_names": "{'max_steps'}", "type": "integer" }, "max_loss": { "title": "Max Loss", "default": 10000, "env_names": "{'max_loss'}", "type": "integer" }, "num_agg": { "title": "Num Agg", "default": 12, "env_names": "{'num_agg'}", "type": "integer" }, "timeout1": { "title": "Timeout1", "default": 30, "env_names": "{'timeout1'}", "type": "integer" }, "timeout2": { "title": "Timeout2", "default": 10, "env_names": "{'timeout2'}", "type": "integer" }, "agg_dir": { "title": "Agg Dir", "default": ".", "env_names": "{'agg_dir'}", "type": "string", "format": "path" }, "published_model_dir": { "title": "Published Model Dir", "env_names": "{'published_model_dir'}", "type": "string", "format": "path" }, "checkpoint_dir": { "title": "Checkpoint Dir", "env_names": "{'checkpoint_dir'}", "type": "string", "format": "path" }, "adios_xml_agg": { "title": "Adios Xml Agg", "env_names": "{'adios_xml_agg'}", "type": "string", "format": "path" }, "reinit": { "title": "Reinit", "default": true, "env_names": "{'reinit'}", "type": "boolean" }, "read_batch": { "title": "Read Batch", "default": 10000, "env_names": "{'read_batch'}", "type": "integer" }, "model": { "title": "Model", "default": "cvae", "env_names": "{'model'}", "type": "string" }, "use_model_checkpoint": { "title": "Use Model Checkpoint", "default": true, "env_names": "{'use_model_checkpoint'}", "type": "boolean" } }, "required": [ "published_model_dir", "checkpoint_dir", "adios_xml_agg" ] }
- Config
extra: str = allow
- Fields
- field adios_xml_agg: pathlib.Path [Required]
- field agg_dir: pathlib.Path = PosixPath('.')
- field batch_size: int = 32
- field checkpoint_dir: pathlib.Path [Required]
- field conv_filter_shapes: List[Tuple[int, int]] = [(3, 3), (3, 3), (3, 3), (3, 3)]
- field conv_filters: List[int] = [64, 64, 64, 64]
- field conv_layers: int = 4
- field conv_strides: List[Tuple[int, int]] = [(1, 1), (2, 2), (1, 1), (1, 1)]
- field dense_dropouts: List[float] = [0.25]
- field dense_layers: int = 1
- field dense_neurons: List[int] = [128]
- field epochs: int = 10
- field final_shape: Tuple[int, ...] = (28, 28, 1)
- field latent_dim: int = 10
- field max_loss: int = 10000
- field max_steps: int = 8000
- field min_step_increment: int = 5000
- field model: str = 'cvae'
- field num_agg: int = 12
- field published_model_dir: pathlib.Path [Required]
- field read_batch: int = 10000
- field reinit: bool = True
- field shuffle: bool = True
- field split_pct: float = 0.8
- field timeout1: int = 30
- field timeout2: int = 10