| NUM_GPUS=1 | |
| MASTER_ADDR=ip-10-0-136-5 | |
| MASTER_PORT=11785 | |
| WORLD_SIZE=1 | |
| PID of this process = 1101409 | |
| ------ 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.001, target='age', num_workers=15, weight_decay=1e-05) | |
| Input dimension: 737280 | |
| total_steps 17400 | |
| wandb_config: | |
| {'model_name': 'HCPflat_raw_age', 'batch_size': 128, 'weight_decay': 1e-05, 'num_epochs': 20, 'seed': 42, 'lr_scheduler_type': 'cycle', 'save_ckpt': False, 'max_lr': 0.001, 'target': 'age', 'num_workers': 15} | |
| wandb_id: HCPflat_raw_beta_age_7cc4d250-ac94-488c-bec4-39b422ee70de | |
| Step [100/870] - Training Loss: 0.7004 - Training MSE: 98.7605 | |
| Step [200/870] - Training Loss: 0.9553 - Training MSE: 104.1519 | |
| Step [300/870] - Training Loss: 1.9018 - Training MSE: 129.4920 | |
| Step [400/870] - Training Loss: 5.3471 - Training MSE: 222.5458 | |
| Step [500/870] - Training Loss: 15.9923 - Training MSE: 460.4371 | |
| Step [600/870] - Training Loss: 32.0503 - Training MSE: 927.5124 | |
| Step [700/870] - Training Loss: 70.9650 - Training MSE: 1574.1360 | |
| Step [800/870] - Training Loss: 80.0552 - Training MSE: 2592.0538 | |