gaia.training package

Subpackages

• gaia.training.engine package
  • Submodules
    • gaia.training.engine.checkpoints module
    • gaia.training.engine.loops module
    • gaia.training.engine.profiler module
    • gaia.training.engine.state module
• gaia.training.solvers package
  • Submodules
    • gaia.training.solvers.inner_solver module
    • gaia.training.solvers.outer_solver module
    • gaia.training.solvers.yoneda_proxy module
• gaia.training.utils package
  • Submodules
    • gaia.training.utils.structure_helpers module
    • gaia.training.utils.validation module

Submodules

• gaia.training.config module
  • OptimizationType
    • OptimizationType.ADAM
    • OptimizationType.ADAMW
    • OptimizationType.SGD
    • OptimizationType.RMSPROP
    • OptimizationType.LION
    • OptimizationType.CATEGORICAL_ADAM
    • OptimizationType.__contains__()
    • OptimizationType.__getitem__()
    • OptimizationType.__iter__()
    • OptimizationType.__len__()
  • SchedulerType
    • SchedulerType.COSINE
    • SchedulerType.LINEAR
    • SchedulerType.EXPONENTIAL
    • SchedulerType.PLATEAU
    • SchedulerType.CATEGORICAL_ANNEALING
    • SchedulerType.__contains__()
    • SchedulerType.__getitem__()
    • SchedulerType.__iter__()
    • SchedulerType.__len__()
  • OptimizationConfig
    • OptimizationConfig.optimizer
    • OptimizationConfig.learning_rate
    • OptimizationConfig.weight_decay
    • OptimizationConfig.momentum
    • OptimizationConfig.betas
    • OptimizationConfig.eps
    • OptimizationConfig.categorical_coherence_weight
    • OptimizationConfig.simplicial_regularization
    • OptimizationConfig.horn_solving_weight
    • OptimizationConfig.scheduler
    • OptimizationConfig.scheduler_params
    • OptimizationConfig.warmup_steps
    • OptimizationConfig.gradient_clip_norm
    • OptimizationConfig.gradient_accumulation_steps
    • OptimizationConfig.__init__()
  • ModelConfig
    • ModelConfig.name
    • ModelConfig.architecture
    • ModelConfig.hidden_dims
    • ModelConfig.activation
    • ModelConfig.dropout
    • ModelConfig.batch_norm
    • ModelConfig.simplicial_depth
    • ModelConfig.categorical_embedding_dim
    • ModelConfig.horn_solver_config
    • ModelConfig.init_method
    • ModelConfig.init_gain
    • ModelConfig.__init__()
  • DataConfig
    • DataConfig.dataset_path
    • DataConfig.batch_size
    • DataConfig.num_workers
    • DataConfig.pin_memory
    • DataConfig.persistent_workers
    • DataConfig.normalize
    • DataConfig.augmentation
    • DataConfig.augmentation_config
    • DataConfig.validation_split
    • DataConfig.test_split
    • DataConfig.random_seed
    • DataConfig.categorical_features
    • DataConfig.simplicial_structure
    • DataConfig.__init__()
  • TrainingConfig
    • TrainingConfig.epochs
    • TrainingConfig.max_steps
    • TrainingConfig.eval_frequency
    • TrainingConfig.save_frequency
    • TrainingConfig.log_level
    • TrainingConfig.log_frequency
    • TrainingConfig.use_tensorboard
    • TrainingConfig.use_wandb
    • TrainingConfig.wandb_project
    • TrainingConfig.checkpoint_dir
    • TrainingConfig.save_top_k
    • TrainingConfig.monitor_metric
    • TrainingConfig.monitor_mode
    • TrainingConfig.early_stopping
    • TrainingConfig.patience
    • TrainingConfig.min_delta
    • TrainingConfig.device
    • TrainingConfig.mixed_precision
    • TrainingConfig.compile_model
    • TrainingConfig.distributed
    • TrainingConfig.world_size
    • TrainingConfig.rank
    • TrainingConfig.categorical_training
    • TrainingConfig.hierarchical_learning
    • TrainingConfig.horn_solving_frequency
    • TrainingConfig.coherence_check_frequency
    • TrainingConfig.model
    • TrainingConfig.data
    • TrainingConfig.optimization
    • TrainingConfig.from_file()
    • TrainingConfig.save()
    • TrainingConfig.__init__()
• gaia.training.hierarchical_message_passing module
• gaia.training.unified_trainer module
  • GAIATrainingConfig
    • GAIATrainingConfig.input_dim
    • GAIATrainingConfig.hidden_dims
    • GAIATrainingConfig.output_dim
    • GAIATrainingConfig.vocab_size
    • GAIATrainingConfig.d_model
    • GAIATrainingConfig.num_heads
    • GAIATrainingConfig.num_layers
    • GAIATrainingConfig.seq_len
    • GAIATrainingConfig.d_ff
    • GAIATrainingConfig.max_seq_length
    • GAIATrainingConfig.learning_rate
    • GAIATrainingConfig.batch_size
    • GAIATrainingConfig.max_epochs
    • GAIATrainingConfig.fuzzy_k_neighbors
    • GAIATrainingConfig.coalgebra_steps
    • GAIATrainingConfig.message_passing_levels
    • GAIATrainingConfig.horn_filling_tolerance
    • GAIATrainingConfig.use_hierarchical_updates
    • GAIATrainingConfig.use_business_units
    • GAIATrainingConfig.use_kan_verification
    • GAIATrainingConfig.verify_coalgebra_dynamics
    • GAIATrainingConfig.log_level
    • GAIATrainingConfig.checkpoint_dir
    • GAIATrainingConfig.log_interval
    • GAIATrainingConfig.__init__()
  • FuzzyDataEncoder
    • FuzzyDataEncoder.__init__()
    • FuzzyDataEncoder.initialize()
    • FuzzyDataEncoder.update()
    • FuzzyDataEncoder.validate()
  • CoalgebraEvolution
    • CoalgebraEvolution.__init__()
    • CoalgebraEvolution.initialize()
    • CoalgebraEvolution.update()
    • CoalgebraEvolution.validate()
  • HierarchicalCommunication
    • HierarchicalCommunication.__init__()
    • HierarchicalCommunication.initialize()
    • HierarchicalCommunication.update()
    • HierarchicalCommunication.validate()
  • KanVerification
    • KanVerification.__init__()
    • KanVerification.initialize()
    • KanVerification.update()
    • KanVerification.validate()
  • GAIATrainer
    • GAIATrainer.__init__()
    • GAIATrainer.train_step()
    • GAIATrainer.validate_step()
    • GAIATrainer.train_epoch()
    • GAIATrainer.train()
    • GAIATrainer.load_checkpoint()
    • GAIATrainer.add_component()
    • GAIATrainer.get_component()
  • create_gaia_trainer()
• gaia.training.verifiers module