NUM_GPUS=1
MASTER_ADDR=ip-10-0-135-126
MASTER_PORT=17074
WORLD_SIZE=1
PID of this process = 2161551
------ 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_81853367-3038-4b91-805f-5066c048cef4
Step [100/435] - Training Loss: 2.2615 - Training Accuracy: 1472.12%
Step [200/435] - Training Loss: 1.4896 - Training Accuracy: 1663.83%
Step [300/435] - Training Loss: 0.8745 - Training Accuracy: 1726.14%
Step [400/435] - Training Loss: 0.5028 - Training Accuracy: 1742.26%
Epoch [1/50] - Training Loss: 1.5308, Training Accuracy: 1743.88% - Validation Loss: 0.5084, Validation Accuracy: 1955.07%
Step [100/435] - Training Loss: 0.2828 - Training Accuracy: 1781.62%
Step [200/435] - Training Loss: 0.3001 - Training Accuracy: 1787.59%
Step [300/435] - Training Loss: 0.2278 - Training Accuracy: 1771.23%
Step [400/435] - Training Loss: 0.2211 - Training Accuracy: 1768.90%
Epoch [2/50] - Training Loss: 0.2600, Training Accuracy: 1767.59% - Validation Loss: 0.2073, Validation Accuracy: 1961.79%