data: | |
data_source: magnus_extracted_positions_m3_pro.pkl | |
preprocessing: [] | |
test_size: 31584 | |
train_size: 102300 | |
val_size: 31654 | |
vocab_size: 945 | |
metrics: | |
best_val_accuracy: 0.0673532570923106 | |
test_accuracy: 0.06648936170212766 | |
test_top3_accuracy: 0.1157548125633232 | |
test_top5_accuracy: 0.14171732522796351 | |
training_time_minutes: 140.37197103500367 | |
model: | |
architecture: AdvancedMagnusModel | |
name: advanced_magnus | |
version: v20250626_170216 | |
training: | |
batch_size: 128 | |
data_source: magnus_extracted_positions_m3_pro.pkl | |
device: mps | |
focal_loss_config: | |
alpha: 0.25 | |
gamma: 2.0 | |
learning_rate: 0.002 | |
min_move_count: 25 | |
model_architecture: AdvancedMagnusModel | |
model_type: AdvancedMagnusModel | |
num_epochs: 50 | |
optimization: advanced_2.5M_params | |
scheduler_config: | |
anneal_strategy: cos | |
max_lr: 0.005 | |
pct_start: 0.1 | |
type: OneCycleLR | |
test_size: 31584 | |
total_params: 2651538 | |
train_size: 102300 | |
val_size: 31654 | |
vocab_size: 945 | |