| [34m[1mwandb[0m: 1 of 1 files downloaded. | |
| Selecting training samples | |
| Keeping 390473 intervals | |
| Selecting validation samples | |
| Keeping 21987 intervals | |
| Selecting test samples | |
| Keeping 22595 intervals | |
| Final sizes: train: (390473, 3), val: (21987, 3), test: (22595, 3) | |
| GPU available: True (cuda), used: True | |
| TPU available: False, using: 0 TPU cores | |
| HPU available: False, using: 0 HPUs | |
| /opt/conda/lib/python3.11/site-packages/pytorch_lightning/loggers/wandb.py:397: There is a wandb run already in progress and newly created instances of `WandbLogger` will reuse this run. If this is not desired, call `wandb.finish()` before instantiating `WandbLogger`. | |
| LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0,1,2,3,4,5,6,7] | |
| Validation DataLoader 0: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 43/43 [00:09<00:00, 4.60it/s] | |
| /opt/conda/lib/python3.11/site-packages/torchmetrics/utilities/prints.py:43: UserWarning: The variance of predictions or target is close to zero. This can cause instability in Pearson correlationcoefficient, leading to wrong results. Consider re-scaling the input if possible or computing using alarger dtype (currently using torch.float32). | |
| LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0,1,2,3,4,5,6,7] | |
| | Name | Type | Params | Mode | |
| ---------------------------------------------------------- | |
| 0 | model | DilatedConvModel | 6.3 M | train | |
| 1 | loss | MSELoss | 0 | train | |
| 2 | activation | Identity | 0 | train | |
| 3 | val_metrics | MetricCollection | 0 | train | |
| 4 | test_metrics | MetricCollection | 0 | train | |
| 5 | transform | Identity | 0 | train | |
| ---------------------------------------------------------- | |
| 6.3 M Trainable params | |
| 0 Non-trainable params | |
| 6.3 M Total params | |
| 25.358 Total estimated model params size (MB) | |
| 131 Modules in train mode | |
| 0 Modules in eval mode | |
| Epoch 14: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 763/763 [07:51<00:00, 1.62it/s, v_num=arkd, train_loss_step=0.551, train_loss_epoch=0.456] | |
| /opt/conda/lib/python3.11/site-packages/torchmetrics/utilities/prints.py:43: UserWarning: The variance of predictions or target is close to zero. This can cause instability in Pearson correlationcoefficient, leading to wrong results. Consider re-scaling the input if possible or computing using alarger dtype (currently using torch.float32). | |
| `Trainer.fit` stopped: `max_epochs=15` reached. | |