deepdrivemd.models.aae.config

pydantic settings deepdrivemd.models.aae.config.AAEModelConfig

Show JSON schema

{
   "title": "AAEModelConfig",
   "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"
      },
      "last_n_h5_files": {
         "title": "Last N H5 Files",
         "default": 10,
         "env_names": "{'last_n_h5_files'}",
         "type": "integer"
      },
      "k_random_old_h5_files": {
         "title": "K Random Old H5 Files",
         "default": 0,
         "env_names": "{'k_random_old_h5_files'}",
         "type": "integer"
      },
      "dataset_name": {
         "title": "Dataset Name",
         "default": "point_cloud",
         "env_names": "{'dataset_name'}",
         "type": "string"
      },
      "rmsd_name": {
         "title": "Rmsd Name",
         "default": "rmsd",
         "env_names": "{'rmsd_name'}",
         "type": "string"
      },
      "fnc_name": {
         "title": "Fnc Name",
         "default": "fnc",
         "env_names": "{'fnc_name'}",
         "type": "string"
      },
      "num_points": {
         "title": "Num Points",
         "default": 3375,
         "env_names": "{'num_points'}",
         "type": "integer"
      },
      "num_features": {
         "title": "Num Features",
         "default": 0,
         "env_names": "{'num_features'}",
         "type": "integer"
      },
      "initial_epochs": {
         "title": "Initial Epochs",
         "default": 10,
         "env_names": "{'initial_epochs'}",
         "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"
      },
      "optimizer_name": {
         "title": "Optimizer Name",
         "default": "Adam",
         "env_names": "{'optimizer_name'}",
         "type": "string"
      },
      "optimizer_lr": {
         "title": "Optimizer Lr",
         "default": 0.0001,
         "env_names": "{'optimizer_lr'}",
         "type": "number"
      },
      "latent_dim": {
         "title": "Latent Dim",
         "default": 64,
         "env_names": "{'latent_dim'}",
         "type": "integer"
      },
      "encoder_filters": {
         "title": "Encoder Filters",
         "default": [
            64,
            128,
            256,
            256,
            512
         ],
         "env_names": "{'encoder_filters'}",
         "type": "array",
         "items": {
            "type": "integer"
         }
      },
      "encoder_kernel_sizes": {
         "title": "Encoder Kernel Sizes",
         "default": [
            5,
            5,
            3,
            1,
            1
         ],
         "env_names": "{'encoder_kernel_sizes'}",
         "type": "array",
         "items": {
            "type": "integer"
         }
      },
      "generator_filters": {
         "title": "Generator Filters",
         "default": [
            64,
            128,
            512,
            1024
         ],
         "env_names": "{'generator_filters'}",
         "type": "array",
         "items": {
            "type": "integer"
         }
      },
      "discriminator_filters": {
         "title": "Discriminator Filters",
         "default": [
            512,
            512,
            128,
            64
         ],
         "env_names": "{'discriminator_filters'}",
         "type": "array",
         "items": {
            "type": "integer"
         }
      },
      "encoder_relu_slope": {
         "title": "Encoder Relu Slope",
         "default": 0.0,
         "env_names": "{'encoder_relu_slope'}",
         "type": "number"
      },
      "generator_relu_slope": {
         "title": "Generator Relu Slope",
         "default": 0.0,
         "env_names": "{'generator_relu_slope'}",
         "type": "number"
      },
      "discriminator_relu_slope": {
         "title": "Discriminator Relu Slope",
         "default": 0.0,
         "env_names": "{'discriminator_relu_slope'}",
         "type": "number"
      },
      "use_encoder_bias": {
         "title": "Use Encoder Bias",
         "default": true,
         "env_names": "{'use_encoder_bias'}",
         "type": "boolean"
      },
      "use_generator_bias": {
         "title": "Use Generator Bias",
         "default": true,
         "env_names": "{'use_generator_bias'}",
         "type": "boolean"
      },
      "use_discriminator_bias": {
         "title": "Use Discriminator Bias",
         "default": true,
         "env_names": "{'use_discriminator_bias'}",
         "type": "boolean"
      },
      "noise_mu": {
         "title": "Noise Mu",
         "default": 0.0,
         "env_names": "{'noise_mu'}",
         "type": "number"
      },
      "noise_std": {
         "title": "Noise Std",
         "default": 1.0,
         "env_names": "{'noise_std'}",
         "type": "number"
      },
      "lambda_rec": {
         "title": "Lambda Rec",
         "default": 0.5,
         "env_names": "{'lambda_rec'}",
         "type": "number"
      },
      "lambda_gp": {
         "title": "Lambda Gp",
         "default": 10,
         "env_names": "{'lambda_gp'}",
         "type": "number"
      },
      "embed_interval": {
         "title": "Embed Interval",
         "default": 1,
         "env_names": "{'embed_interval'}",
         "type": "integer"
      },
      "tsne_interval": {
         "title": "Tsne Interval",
         "default": 5,
         "env_names": "{'tsne_interval'}",
         "type": "integer"
      },
      "sample_interval": {
         "title": "Sample Interval",
         "default": 20,
         "env_names": "{'sample_interval'}",
         "type": "integer"
      },
      "num_data_workers": {
         "title": "Num Data Workers",
         "default": 0,
         "env_names": "{'num_data_workers'}",
         "type": "integer"
      },
      "dataset_location": {
         "title": "Dataset Location",
         "default": "storage",
         "env_names": "{'dataset_location'}",
         "type": "string"
      },
      "wandb_project_name": {
         "title": "Wandb Project Name",
         "env_names": "{'wandb_project_name'}",
         "type": "string"
      }
   }
}

Config

• extra: str = allow

Fields

• batch_size (int)

• dataset_location (str)

• dataset_name (str)

• discriminator_filters (List[int])

• discriminator_relu_slope (float)

• embed_interval (int)

• encoder_filters (List[int])

• encoder_kernel_sizes (List[int])

• encoder_relu_slope (float)

• epochs (int)

• fnc_name (str)

• generator_filters (List[int])

• generator_relu_slope (float)

• initial_epochs (int)

• k_random_old_h5_files (int)

• lambda_gp (float)

• lambda_rec (float)

• last_n_h5_files (int)

• latent_dim (int)

• noise_mu (float)

• noise_std (float)

• num_data_workers (int)

• num_features (int)

• num_points (int)

• optimizer_lr (float)

• optimizer_name (str)

• rmsd_name (str)

• sample_interval (int)

• tsne_interval (int)

• use_discriminator_bias (bool)

• use_encoder_bias (bool)

• use_generator_bias (bool)

• wandb_project_name (Optional[str])

field batch_size: int = 32

field dataset_location: str = 'storage'

field dataset_name: str = 'point_cloud'

field discriminator_filters: List[int] = [512, 512, 128, 64]

field discriminator_relu_slope: float = 0.0

field embed_interval: int = 1

field encoder_filters: List[int] = [64, 128, 256, 256, 512]

field encoder_kernel_sizes: List[int] = [5, 5, 3, 1, 1]

field encoder_relu_slope: float = 0.0

field epochs: int = 10

field fnc_name: str = 'fnc'

field generator_filters: List[int] = [64, 128, 512, 1024]

field generator_relu_slope: float = 0.0

field initial_epochs: int = 10

field k_random_old_h5_files: int = 0

field lambda_gp: float = 10

field lambda_rec: float = 0.5

field last_n_h5_files: int = 10

field latent_dim: int = 64

field noise_mu: float = 0.0

field noise_std: float = 1.0

field num_data_workers: int = 0

field num_features: int = 0

field num_points: int = 3375

field optimizer_lr: float = 0.0001

field optimizer_name: str = 'Adam'

field rmsd_name: str = 'rmsd'

field sample_interval: int = 20

field tsne_interval: int = 5

field use_discriminator_bias: bool = True

field use_encoder_bias: bool = True

field use_generator_bias: bool = True

field wandb_project_name: Optional[str] = PydanticUndefined