SSB_Screening_Model / config_resolved.yaml
Allanatrix's picture
Upload config_resolved.yaml with huggingface_hub
8370779 verified
project:
slug: ssb_rtx6000x2
name: SSB RTX6000x2
owner: null
experiment:
name: ssb_npz_rtx6000x2_run
output_dir: artifacts/ssb_rtx6000x2/runs
tags:
provider: local
gpu: rtx6000x2
dataset: ssb_npz_v1
notes: null
data:
dataset_name: npz_classification
dataset_version: ssb_npz_v1
source_uri: data/processed/ssb_npz_v1
batch_size: 305
num_workers: 8
shuffle: true
pin_memory: true
split:
train: 0.8
validation: 0.1
test: 0.1
preprocessing:
class_weighted_loss: true
augmentations: []
cache_dir: data/cache
model:
name: mlp_classifier
parameters:
input_dim: 144
hidden_dims:
- 512
- 256
- 128
num_classes: 3
dropout: 0.226947161178787
args: null
pretrained: false
checkpoint_path: null
training:
epochs: 45
max_steps: null
log_every_n_steps: 25
val_every_n_steps: null
gradient_clip_norm: 1.0
optimizer:
name: adamw
lr: 0.0002547762821605831
weight_decay: 0.010098717791485455
eps: 1.0e-08
betas: null
args: {}
scheduler:
name: cosine
args:
t_max: 43
early_stop_patience: 0
distributed:
backend: nccl
world_size: 2
num_nodes: 1
node_rank: 0
master_addr: localhost
master_port: 29500
gradient_accumulation_steps: 4
mixed_precision: true
seed: 2026
checkpoint:
dir: artifacts/ssb_rtx6000x2/checkpoints
monitor: val_loss
mode: min
save_top_k: 2
save_every_n_epochs: 2
save_last: true
resume_from: null
logging:
level: INFO
log_dir: logs/ssb_rtx6000x2
tensorboard: true
mlflow_tracking_uri: null
mlflow_experiment: null
flush_steps: 50
timeout_seconds: null
evaluation:
metrics:
- accuracy
- f1
- precision
- recall
batch_size: 1024
save_predictions: false
output_dir: artifacts/ssb_rtx6000x2/eval
generate_confusion_matrix: true
generate_calibration_plot: false
calibration_bins: 10