NUM_GPUS=1
MASTER_ADDR=ip-10-0-135-126
MASTER_PORT=13292
WORLD_SIZE=1
PID of this process = 2152625
------ ARGS ------- 
 Namespace(model_suffix='beta', hcp_flat_path='/weka/proj-medarc/shared/HCP-Flat', batch_size=256, wandb_log=True, num_epochs=50, lr_scheduler_type='cycle', save_ckpt=False, seed=42, max_lr=1e-05, target='trial_type', num_workers=15, weight_decay=1e-05)
Number of classes: 21
Input dimension: 737280
total_steps 21750
wandb_config:
 {'model_name': 'HCPflat_raw_trial_type', 'batch_size': 256, 'weight_decay': 1e-05, 'num_epochs': 50, 'seed': 42, 'lr_scheduler_type': 'cycle', 'save_ckpt': False, 'max_lr': 1e-05, 'target': 'trial_type', 'num_workers': 15}
wandb_id: HCPflat_raw_beta_trial_type_3d482888-f4eb-482f-a3c2-2056642d97e2
wandb: 🚀 View run HCPflat_raw_beta_trial_type at: https://stability.wandb.io/ckadirt/fMRI-foundation-model/runs/HCPflat_raw_beta_trial_type_3d482888-f4eb-482f-a3c2-2056642d97e2
wandb: Find logs at: wandb/run-20241127_015933-HCPflat_raw_beta_trial_type_3d482888-f4eb-482f-a3c2-2056642d97e2/logs