|
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% |
|
|