neurovlm.training.AutoencoderTrainConfig

neurovlm.training.AutoencoderTrainConfig#

class neurovlm.training.AutoencoderTrainConfig(output_root='runs', run_id=None, variant='mixed_baseline', domain=None, seed=42, device='auto', epochs=100, batch_size=64, eval_batch_size=None, num_workers=0, learning_rate=0.0003, weight_decay=0.0001, gradient_clip=1.0, amp=True, early_stopping_patience=10, early_stopping_min_delta=0.0, include_voxel_auroc=False, max_train_batches=None, max_eval_batches=None, limit=None, split_dir=None, volume_path=None, initialization='auto', from_run=None, init_checkpoint=None, resume=None, preset='retained_base64_v1', target_shape=(36, 45, 38), in_channels=1, base_channels=64, num_blocks=4, latent_dim=384, dropout=0.1, norm='group', pooling='max')[source]#

Typed configuration for mixed pretraining or domain fine-tuning.

initialization="auto" means scratch initialization for the mixed baseline and the released mixed autoencoder for a fine-tuned run. Released resources remain the default; split_dir and volume_path are explicit local overrides.

Parameters:

  • output_root (str | Path)

  • run_id (str | None)

  • variant (Literal['mixed_baseline', 'finetuned'])

  • domain (Literal['pubmed', 'nilearn', 'neurovault'] | None)

  • seed (int)

  • device (str)

  • epochs (int)

  • batch_size (int)

  • eval_batch_size (int | None)

  • num_workers (int)

  • learning_rate (float)

  • weight_decay (float)

  • gradient_clip (float | None)

  • amp (bool)

  • early_stopping_patience (int | None)

  • early_stopping_min_delta (float)

  • include_voxel_auroc (bool)

  • max_train_batches (int | None)

  • max_eval_batches (int | None)

  • limit (int | None)

  • split_dir (str | Path | None)

  • volume_path (str | Path | None)

  • initialization (Literal['auto', 'scratch', 'released_mixed'])

  • from_run (str | Path | None)

  • init_checkpoint (str | Path | None)

  • resume (str | Path | None)

  • preset (Literal['retained_base64_v1', 'custom'])

  • target_shape (tuple[int, int, int])

  • in_channels (int)

  • base_channels (int)

  • num_blocks (int)

  • latent_dim (int)

  • dropout (float)

  • norm (Literal['group', 'batch', 'instance', 'none'])

  • pooling (Literal['max', 'stride'])

__init__(output_root='runs', run_id=None, variant='mixed_baseline', domain=None, seed=42, device='auto', epochs=100, batch_size=64, eval_batch_size=None, num_workers=0, learning_rate=0.0003, weight_decay=0.0001, gradient_clip=1.0, amp=True, early_stopping_patience=10, early_stopping_min_delta=0.0, include_voxel_auroc=False, max_train_batches=None, max_eval_batches=None, limit=None, split_dir=None, volume_path=None, initialization='auto', from_run=None, init_checkpoint=None, resume=None, preset='retained_base64_v1', target_shape=(36, 45, 38), in_channels=1, base_channels=64, num_blocks=4, latent_dim=384, dropout=0.1, norm='group', pooling='max')#
Parameters:

  • output_root (str | Path)

  • run_id (str | None)

  • variant (Literal['mixed_baseline', 'finetuned'])

  • domain (Literal['pubmed', 'nilearn', 'neurovault'] | None)

  • seed (int)

  • device (str)

  • epochs (int)

  • batch_size (int)

  • eval_batch_size (int | None)

  • num_workers (int)

  • learning_rate (float)

  • weight_decay (float)

  • gradient_clip (float | None)

  • amp (bool)

  • early_stopping_patience (int | None)

  • early_stopping_min_delta (float)

  • include_voxel_auroc (bool)

  • max_train_batches (int | None)

  • max_eval_batches (int | None)

  • limit (int | None)

  • split_dir (str | Path | None)

  • volume_path (str | Path | None)

  • initialization (Literal['auto', 'scratch', 'released_mixed'])

  • from_run (str | Path | None)

  • init_checkpoint (str | Path | None)

  • resume (str | Path | None)

  • preset (Literal['retained_base64_v1', 'custom'])

  • target_shape (tuple[int, int, int])

  • in_channels (int)

  • base_channels (int)

  • num_blocks (int)

  • latent_dim (int)

  • dropout (float)

  • norm (Literal['group', 'batch', 'instance', 'none'])

  • pooling (Literal['max', 'stride'])

Return type:

None

Methods

__init__([output_root, run_id, variant, ...])

architecture()

Attributes

amp

base_channels

batch_size

device

domain

dropout

early_stopping_min_delta

early_stopping_patience

epochs

eval_batch_size

from_run

gradient_clip

in_channels

include_voxel_auroc

init_checkpoint

initialization

latent_dim

learning_rate

limit

max_eval_batches

max_train_batches

metric_direction

norm

num_blocks

num_workers

output_root

pooling

preset

primary_metric

resume

run_id

seed

split_dir

target_shape

variant

volume_path

weight_decay