NUM_GPUS=1
MASTER_ADDR=ip-10-0-136-246
MASTER_PORT=17060
WORLD_SIZE=1
PID of this process = 564369
------ ARGS ------- 
 Namespace(model_suffix='beta', hcp_flat_path='/weka/proj-medarc/shared/HCP-Flat', batch_size=128, wandb_log=True, num_epochs=20, lr_scheduler_type='cycle', save_ckpt=False, seed=42, max_lr=0.1, target='sex', num_workers=15, weight_decay=1e-05)
Input dimension: 737280